jax.numpy.right_shiftΒΆ

jax.numpy.right_shift(x1, x2)[source]ΒΆ

Shift the bits of an integer to the right.

LAX-backend implementation of right_shift().

Original docstring below.

Bits are shifted to the right x2. Because the internal representation of numbers is in binary format, this operation is equivalent to dividing x1 by 2**x2.

Parameters
  • x1 (array_like, int) – Input values.

  • x2 (array_like, int) – Number of bits to remove at the right of x1. If x1.shape != x2.shape, they must be broadcastable to a common shape (which becomes the shape of the output).

Returns

out – Return x1 with bits shifted x2 times to the right. This is a scalar if both x1 and x2 are scalars.

Return type

ndarray, int