Проверьте, равны ли тензоры PyTorch в эпсилоне

Question

Как проверить, являются ли два тензора PyTorch семантически равными?

Учитывая ошибки с плавающей запятой, я хочу знать, отличаются ли элементы только небольшим значением эпсилона.

Answer 1

На момент написания данной статьи это недокументированная функция в последней стабильной версии (0.4.1), но документация находится в ветке master (unstable).

torch.allclose() вернет логическое значение, указывающее, равны ли все поэлементные различия с учетом погрешности.

Кроме того, имеется недокументированное isclose():

>>> torch.isclose(torch.Tensor([1]), torch.Tensor([1.00000001]))
tensor([1], dtype=torch.uint8)