一、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)