Python PyTorch allclose用法及代碼示例

本文簡要介紹python語言中 torch.allclose 的用法。

用法:

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

參數：

• input(Tensor) -第一個要比較的張量

• other(Tensor) -要比較的第二張量

• atol(float,可選的) -絕對寬容。默認值：1e-08

• rtol(float,可選的) -相對公差。默認值：1e-05

• equal_nan(bool,可選的) -如果 True ，則兩個 NaN 將被視為相等。默認值：False

此函數檢查是否所有 input 和 other 都滿足條件：

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

元素層麵，對於 input 和 other 的所有元素。該函數的行為類似於numpy.allclose

例子：

>>> 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