mlx.core.equal

equal(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array

逐元素相等。

对两个具有 numpy 风格广播语义的数组进行相等比较。输入数组可以是标量，也可以两个都是标量。

参数:

• a (array) – 输入数组或标量。

• b (array) – 输入数组或标量。

返回值:

逐元素比较 a == b。

返回类型:

array