torch.eq¶
- torch.eq(input, other, *, out=None) Tensor ¶
Computes element-wise equality
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 equal toother
and False elsewhere
Example:
>>> torch.eq(torch.tensor([[1, 2], [3, 4]]), torch.tensor([[1, 1], [4, 4]])) tensor([[ True, False], [False, True]])