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

Question

对于 Pytorch 张量A ：

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

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

谢谢，

Answer 1

torch.Tensor实现__contains__ 。 所以，你可以使用：

1 in A

如果元素1在A ，则返回True否则返回False 。