mlx.core.equal

目录

mlx.core.equal#

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

Element-wise equality.

Equality comparison on two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

参数:
  • a (array) -- Input array or scalar.

  • b (array) -- Input array or scalar.

返回:

The element-wise comparison a == b.

返回类型:

array