torch.allclose

torch.allclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False) → bool

Эта функция проверяет, удовлетворяют ли все input и other условию:

$$\lvert \text{input} - \text{other} \rvert \leq \texttt{atol} + \texttt{rtol} \times \lvert \text{other} \rvert$$

поэлементно для всех элементов input и other. Поведение этой функции аналогично numpy.allclose

Параметры:

• input (Тензор) – первый тензор для сравнения
• other (Тензор) – второй тензор для сравнения
• atol (float, необязательно) – абсолютная погрешность. По умолчанию: 1e-08
• rtol (float, необязательно) – относительная погрешность. По умолчанию: 1e-05
• equal_nan (bool, необязательно) – если True, то два NaN будут считаться равными. По умолчанию: False

Пример:

>>> torch.allclose(torch.tensor([10000., 1e-07]), torch.tensor([10000.1, 1e-08]))
False
>>> torch.allclose(torch.tensor([10000., 1e-08]), torch.tensor([10000.1, 1e-09]))
True
>>> torch.allclose(torch.tensor([1.0, float('nan')]), torch.tensor([1.0, float('nan')]))
False
>>> torch.allclose(torch.tensor([1.0, float('nan')]), torch.tensor([1.0, float('nan')]), equal_nan=True)
True