torch.equal

torch.equal(input, other) → bool

如果兩個張量具有相同的大小和元素，則為True，否則為False。

注意

包含 NaN 的張量之間永遠不相等。此外，此函式在比較期間不區分張量的資料型別。要進行更徹底的張量檢查，請使用 torch.testing.assert_close()。

示例

>>> torch.equal(torch.tensor([1, 2]), torch.tensor([1, 2]))
True
>>> torch.equal(torch.tensor([3, torch.nan]), torch.tensor([3, torch.nan]))
False
>>> torch.equal(torch.tensor([1, 2, 3], dtype=torch.int32), torch.tensor([1, 2, 3], dtype=torch.float32))
True