torch.isclose

torch.isclose是 PyTorch 中的一个函数，用于判断两个张量中的对应元素是否接近相等。

其函数签名为：torch.isclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False)。

参数说明：

• input 和 other：要进行比较的两个张量。
• rtol（相对容差）：相对容差参数，默认值为1e-05。相对容差是相对于两个数中较大数的允许相对误差。例如，如果两个数分别为a和b，则它们在相对容差范围内接近相等的条件是abs(a - b) <= max(rtol * max(abs(a), abs(b)), atol)。
• atol（绝对容差）：绝对容差参数，默认值为1e-08。绝对容差是一个固定的允许绝对误差。
• equal_nan：布尔值参数，指示是否将两个NaN值视为接近相等。默认值为False。

返回值是一个布尔类型的张量，其中每个元素表示对应位置的两个输入张量中的元素是否接近相等。