jax.lax.shift_right_arithmetic¶

jax.lax.shift_right_arithmetic(x, y)[source]¶

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

Parameters

  • x (Any) –

  • y (Any) –

Return type

Any