torch.isclose¶
- torch.isclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False) Tensor ¶
返回一个新的张量,其中布尔元素表示
input
的每个元素是否“接近”other
的相应元素。接近度定义为其中
input
和other
是有限的。当input
和/或other
非有限时,当且仅当它们相等时它们才接近,当equal_nan
为 True 时,NaN 被认为彼此相等。- 参数
示例
>>> 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])