Как проверить, есть ли определенное число в тензоре Pytorch?

Question

для тензора Pytorch A:

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

есть ли способ проверить, является ли число 1 элементом тензора A? Например, есть ли функция pytorch, которая возвращает True, если 1 является элементом A, и возвращает False, если 1 не является элементом A?

Спасибо,

Answer 1

torch.Tensor реализует __contains__. Таким образом, вы можете просто использовать:

1 in A

Возвращает True, если элемент 1 находится в A, и False в противном случае.