PyTorch基础（六）----- torch.eq()方法

一、torch.eq()方法详解

对两个张量Tensor进行逐元素的比较，若相同位置的两个元素相同，则返回True；若不同，返回False。

torch.eq(input, other, *, out=None)

• Parameters（参数）:
  • input ：必须是一个Tensor，该张量用于比较
  • other ：可以是一个张量Tensor，也可以是一个值value
• return（返回值）：返回一个Boolean类型的张量，对两个张量Tensor进行逐元素的比较，若相同位置的两个元素相同，则返回True；若不同，返回False。

二、案例

x = torch.tensor([[1, 2], [3, 4]])
y = torch.tensor([[1, 1], [3, 3]])
print(x)
print(y)
out = torch.eq(x, y)
print(out)