快捷方式

torch.isclose

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

返回一个新的张量,其中布尔元素表示 input 的每个元素是否“接近” other 的相应元素。接近度定义为

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

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

参数
  • input (Tensor) – 要比较的第一个张量

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

  • atol (float, optional) – 绝对公差。默认值:1e-08

  • rtol (float, optional) – 相对公差。默认值:1e-05

  • equal_nan (bool, optional) – 如果 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])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取适合初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源