jax_left_shift

probly.representation.jax_functions.jax_left_shift(x1: JaxArrayLike, x2: JaxArrayLike, /) jax.Array[source]
probly.representation.jax_functions.jax_left_shift(x1: JaxOperand, x2: JaxOperand, /) object

Shift bits left elementwise, mirroring jax.numpy.left_shift.

Parameters:
  • x1 – Integer values to shift.

  • x2 – Number of bit positions to shift each value.

Returns:

The left-shifted values, or the result supplied by a custom override.