torch.equal

torch.equal(input, other) → bool

如果两个张量的大小和元素相同，则为 True，否则为 False。

示例

>>> torch.equal(torch.tensor([1, 2]), torch.tensor([1, 2]))
True