mlx.core.array_equal

array_equal(a: scalar | array, b: scalar | array, equal_nan: bool = False, stream: None | Stream | Device = None) → array

陣列相等性檢查。

比較兩個陣列是否相等。只有在陣列形狀相同且值相等時才回傳 True。陣列不必具有相同型別也可視為相等。

參數:

• a (array) -- 輸入陣列或純量。

• b (array) -- 輸入陣列或純量。

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

回傳:

布林純量陣列。

回傳型別:

array