mlx.core.allclose#

allclose(a: array, b: array, /, rtol: float = 1e-05, atol: float = 1e-08, *, equal_nan: bool = False, stream: None | Stream | Device = None) → array#

对两个数组进行近似比较。

无穷大值如果符号相同则视为相等，NaN 值除非 equal_nan 为 True 否则不相等。

如果满足以下条件，则认为数组相等：

all(abs(a - b) <= (atol + rtol * abs(b)))

注意，与 array_equal() 不同，此函数支持 numpy 风格的广播。

参数:

• a (数组) – 输入数组。

• b (数组) – 输入数组。

• rtol (float) – 相对容差。

• atol (float) – 绝对容差。

• equal_nan (bool) – 如果为 True，NaNs 视为相等。默认为 False。

返回:

表示数组是否接近的布尔输出标量。

返回类型:

数组