jax.lax.eq

Contents

jax.lax.eq#

jax.lax.eq(x, y)[source]#

Elementwise equals: \(x = y\).

Parameters:
  • x (jax.typing.ArrayLike)

  • y (jax.typing.ArrayLike)

Return type:

Array