张量相等和布尔值作为返回值

Question

所以，我在 SO 上关注了这个 answer

我试图将两个张量等同起来

torch.equal(x_valid[0], x_valid[:1]) 返回False 而 torch.all(torch.eq(x_valid[0], x_valid[:1])) 返回tensor(1, dtype=torch.uint8)

我知道两个张量都与x_valid 的第一个值相同，那么为什么torch.equal 返回False？

除了x_valid[0] 返回([0, 0, ...,0]) 和x_valid[:1] 返回([[0, 0, ...,0]])这一事实之外

但是它们的类型仍然是tensor。所以我真的不明白为什么第一个查询的输出是False

Answer 1

torch.equal(tensor1, tensor2) 如果两个张量具有相同的大小和元素，则返回 True，否则返回 False。检查here。

例子：

y = torch.tensor([[0, 0, 0]])
print(y[0], y[0].shape)
print(y[:1], y[:1].shape)
print(torch.equal(y[0], y[:1]))
print(torch.equal(y[0], y[:1][0])) # (torch.Size([3]), torch.Size([3]))

输出：

tensor([0, 0, 0]) torch.Size([3])
tensor([[0, 0, 0]]) torch.Size([1, 3])
False
True

而torch.eq(input, other, out=None) 计算元素相等性。这里需要注意的是，第二个参数可以是数字或张量，其形状为broadcastable，第一个参数。