cupy.allclose

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

如果两个数组在容差范围内逐元素相等，则返回 True。

当满足以下方程时，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。

返回值：

一个布尔型 0 维数组。如果其值为 True，则两个数组在容差范围内逐元素相等。

返回类型：

cupy.ndarray

另请参阅

numpy.allclose()