jax.lax.shift_right_logical

jax.lax.shift_right_logical#

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

Elementwise logical right shift: \(x \gg y\).

Parameters:
  • x (jax.typing.ArrayLike)

  • y (jax.typing.ArrayLike)

Return type:

Array