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 (Tensor) – 需要比较的首个张量

• other (Tensor) – 需要进行比较的第二个张量

• 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