torch.isclose

torch.isclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False) → 张量

返回一个包含布尔元素的新张量，表示input中的每个元素是否“接近”other中对应的元素。 接近度定义为：

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

其中 input 和 other 是有限的。当 input 和/或 other 是非有限时，它们接近当且仅当它们相等，当 equal_nan 为 True 时，NaN 被视为彼此相等。

Parameters

• 输入 (张量) – 第一个比较的张量

• 其他 (Tensor) – 比较的第二个张量

• atol (float, 可选) – 绝对容差。默认值：1e-08

• rtol (float, 可选) – 相对容差。默认值：1e-05

• equal_nan (布尔值, 可选) – 如果True，则两个NaN将被视为相等。默认值：False

示例：

>>> torch.isclose(torch.tensor((1., 2, 3)), torch.tensor((1 + 1e-10, 3, 4)))
tensor([ True, False, False])
>>> torch.isclose(torch.tensor((float('inf'), 4)), torch.tensor((float('inf'), 6)), rtol=.5)
tensor([True, True])