mlx.core.isclose

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

Returns a boolean array where two arrays are element-wise equal within a tolerance.

無限值若符號相同則視為相等；NaN 值除非 equal_nan 為 True，否則視為不相等。

Two values are considered equal if:

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

注意：不同於 array_equal()，此函式支援 numpy 風格廣播。

參數:

• a (array) -- Input array.

• b (array) -- Input array.

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

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

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

回傳:

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

回傳型別:

array