当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch isclose用法及代码示例


本文简要介绍python语言中 torch.isclose 的用法。

用法:

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

参数

  • input(Tensor) -第一个要比较的张量

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

  • atol(float,可选的) -绝对宽容。默认值:1e-08

  • rtol(float,可选的) -相对公差。默认值:1e-05

  • equal_nan(bool,可选的) -如果 True ,则两个 NaN 将被视为相等。默认值:False

返回一个带有布尔元素的新张量,表示 input 的每个元素是否为 “close” 到 other 的相应元素。紧密度定义为:

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

例子:

>>> 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.org大神的英文原创作品 torch.isclose。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。