jax_right_shift¶
- probly.representation.jax_functions.jax_right_shift(x1: JaxArrayLike, x2: JaxArrayLike, /) jax.Array[source]¶
- probly.representation.jax_functions.jax_right_shift(x1: JaxOperand, x2: JaxOperand, /) object
Shift bits right elementwise, mirroring
jax.numpy.right_shift.- Parameters:
x1 – Integer values to shift.
x2 – Number of bit positions to shift each value.
- Returns:
The right-shifted values, or the result supplied by a custom override.