jax.lax.shift_right_arithmetic# jax.lax.shift_right_arithmetic(x, y)[source]# Elementwise arithmetic right shift: \(x \gg y\). Parameters: x (jax.typing.ArrayLike) y (jax.typing.ArrayLike) Return type: Array