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 (array) -- 輸入陣列。

• b (array) -- 輸入陣列。

• rtol (float) -- 相對容差。

• atol (float) -- 絕對容差。

• equal_nan (bool) -- 若為 True，NaN 會視為相等。預設為 False。

回傳:

布林輸出純量，表示陣列是否相近。

回傳型別:

array