PyTorch
torch / torch
torch.ne¶
-
torch.
ne
(input, other, *, out=None) → Tensor¶ Computes input=other element-wise.
The second argument can be a number or a tensor whose shape is broadcastable with the first argument.
- Parameters
- Keyword Arguments
out (Tensor, optional) – the output tensor.
- Returns
A boolean tensor that is True where
input
is not equal toother
and False elsewhere
Example:
>>> torch.ne(torch.tensor([[1, 2], [3, 4]]), torch.tensor([[1, 1], [4, 4]])) tensor([[False, True], [True, False]])
此页内容是否对您有帮助
感谢反馈!