cupy.isclose

cupy.isclose(a, b, rtol=1e-05, atol=1e-08, equal_nan=False)

返回一个布尔数组，表示两个数组在容差范围内是否相等。

当以下等式成立时，a 和 b 中的两个值被认为是相等的。

\[|a - b| \le \mathrm{atol} + \mathrm{rtol} |b|\]

参数：

• a (cupy.ndarray) – 要比较的输入数组。

• b (cupy.ndarray) – 要比较的输入数组。

• rtol (float) – 相对容差。

• atol (float) – 绝对容差。

• equal_nan (bool) – 如果为 True，则 a 中的 NaN 将被视为与 b 中的 NaN 相等。

返回值：

一个布尔数组，存储 a 和 b 相等的位置。

返回类型：

cupy.ndarray

另请参阅

numpy.isclose()