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 的相應元素。緊密度定義為：

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

其中input 和other 是有限的。其中 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])