python - 如何检查某个数字是否在 Pytorch 张量中？

Question

对于 Pytorch 张量 A :

A = tensor([1,0,0],
           [0,0,0])

有什么方法可以检查数字 1 是否是张量的元素 A ?
比如是否有一个返回 True 的 pytorch 函数是 1 是 A 的一个元素, 如果 1 不是 A 的元素，则返回 False ?

谢谢，

Answer 1

torch.Tensor工具__contains__ .所以，你可以使用:

1 in A

这将返回 True如果元素 1在 A , 和 False除此以外。