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