mindspore.ops.IsClose
- class mindspore.ops.IsClose(rtol=1e-05, atol=1e-08, equal_nan=False)[源代码]
返回一个bool型Tensor,表示 input 的每个元素与 other 的每个元素在给定容忍度内是否“接近”。其中“接近”的数学公式为:
\[|input-other| ≤ atol + rtol × |other|\]更多参考详见
mindspore.ops.isclose()
。- 参数:
rtol (float, 可选) - 相对容忍度。默认值:
1e-05
。atol (float, 可选) - 绝对容忍度。默认值:
1e-08
。equal_nan (bool, 可选) - 若为True,则两个NaN被视为相同。默认值:
False
。
- 输入:
input (Tensor) - 对比的第一个输入。
other (Tensor) - 对比的第二个输入。
- 输出:
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]