mindspore.ops.IsClose

class mindspore.ops.IsClose(rtol=1e-05, atol=1e-08, equal_nan=True)[源代码]

返回一个bool型Tensor，表示 x1 的每个元素与 x2 的每个元素在给定容忍度内是否“接近”。

更多参考详见 mindspore.ops.isclose()。

参数：

  • rtol (float, 可选) - 相对容忍度。默认值： 1e-05 。

  • atol (float, 可选) - 绝对容忍度。默认值： 1e-08 。

  • equal_nan (bool, 可选) - 若为True，则两个NaN被视为相同。默认值： True 。

输入：

  • input (Tensor) - 对比的第一个输入，支持的类型有float32、float16、int32。

  • other (Tensor) - 对比的第二个输入，支持的类型有float32、float16、int32。

输出：

Tensor，shape与 input 和 other 广播后的shape相同，数据类型是bool。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore.ops import IsClose
>>> input = Tensor(np.array([1.3, 2.1, 3.2, 4.1, 5.1]), mindspore.float16)
>>> other = Tensor(np.array([1.3, 3.3, 2.3, 3.1, 5.1]), mindspore.float16)
>>> isclose = IsClose()
>>> output = isclose(input, other)
>>> print(output)
[ True False False False  True]