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