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