mlx.core.equal

目录

mlx.core.equal#

相等(a 标量 | 数组, b 标量 | 数组, None | Stream | Device = ) 数组#

元素级别的相等性。

在两个数组上进行相等比较,使用numpy风格的广播语义。 输入数组中的一个或两个也可以是标量。

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

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

Returns

元素级别的比较 a == b

Return type

数组