Эквивалент Pytorch для логики Nandy и родственников?

Question

Имеет ли Pytorch эквивалент элементарных логических операторов Numpy (logical_and, logical_or, logical_not и logical_xor)?Вызов функций Numpy для тензоров Pytorch, кажется, работает достаточно хорошо при использовании процессора, даже создавая тензор Pytorch в качестве вывода.Я в основном спрашиваю, потому что полагаю, что это не сработает так хорошо, если вычисление pytorch будет выполняться в GPU.

Я просмотрел Индекс документации Pytorch по всем функциям, содержащим строку "и"и ни один не кажется уместным.

Answer 1

PyTorch поддерживает логические операции на ByteTensor.Вы можете использовать логические операции, используя операторы &, |, ^, ~ следующим образом:

>>> a = torch.ByteTensor([0, 1, 1, 0])
>>> b = torch.ByteTensor([1, 1, 0, 0])

>>> a & b  # logical and
tensor([0, 1, 0, 0], dtype=torch.uint8)

>>> a | b  # logical or
tensor([1, 1, 1, 0], dtype=torch.uint8)

>>> a ^ b  # logical xor
tensor([1, 0, 1, 0], dtype=torch.uint8)

>>> ~a  # logical not
tensor([1, 0, 0, 1], dtype=torch.uint8)