jax.lax.shift_left#

jax.lax.shift_left(x,y)[source]#

Elementwise left shift:\(x \ll y\).

This function lowers directly to thestablehlo.shift_left operation.

Parameters:

• x (ArrayLike) – Input arrays. Must have matching integer dtypes. If neither is ascalar,x andy must have the same number of dimensions andbe broadcast compatible.

• y (ArrayLike) – Input arrays. Must have matching integer dtypes. If neither is ascalar,x andy must have the same number of dimensions andbe broadcast compatible.

Returns:

An array of the same dtype asx andy containing the element-wiseleft shift of each pair of broadcasted entries.

Return type:

Array

See also

• jax.numpy.left_shift(): NumPy wrapper for this API, also accessiblevia thex<<y operator on JAX arrays.

• jax.lax.shift_right_arithmetic(): Elementwise arithmetic right shift.

• jax.lax.shift_right_logical(): Elementwise logical right shift.