torch.allclose

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

此函数检查 input 和 other 是否满足以下条件：

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

元素级，对于 input 和 other 的所有元素。此函数的行为类似于 numpy.allclose

参数:

• 输入（张量）- 要比较的第一个张量

• 其他（张量）- 要比较的第二个张量

• atol（浮点数，可选）- 绝对容差。默认：1e-08

• rtol（浮点数，可选）- 相对容差。默认：1e-05

• equal_nan（布尔值，可选）- 如果 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