jax.numpy.equal

Contents

jax.numpy.equal#

jax.numpy.equal(x, y, /)[source]#

Return (x1 == x2) element-wise.

LAX-backend implementation of numpy.equal().

Original docstring below.

Returns:

out – Output array, element-wise comparison of x1 and x2. Typically of type bool, unless dtype=object is passed. This is a scalar if both x1 and x2 are scalars.

Return type:

ndarray or scalar

Parameters:
  • x (jax.typing.ArrayLike)

  • y (jax.typing.ArrayLike)