jax.lax.shift_left

jax.lax.shift_left(x, y)

逐元素左移：\(x \ll y\)。

此函数直接降低到 stablehlo.shift_left 操作。

参数:

• x (ArrayLike) – 输入数组。必须具有匹配的整数 dtype。如果都不是标量，则 x 和 y 必须具有相同的维度数并且是广播兼容的。

• y (ArrayLike) – 输入数组。必须具有匹配的整数 dtype。如果都不是标量，则 x 和 y 必须具有相同的维度数并且是广播兼容的。

返回值:

与 x 和 y 具有相同 dtype 的数组，包含每对广播条目的逐元素左移结果。

返回类型:

Array

另请参阅

• jax.numpy.left_shift()：此 API 的 NumPy 包装器，也可以通过 JAX 数组上的 x << y 运算符访问。

• jax.lax.shift_right_arithmetic()：逐元素算术右移。

• jax.lax.shift_right_logical()：逐元素逻辑右移。