jax.lax.shift_right_arithmetic

jax.lax.shift_right_arithmetic(x, y)

按元素的算术右移：\(x \gg y\)。

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

参数:

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

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

返回:

一个与 x 和 y 具有相同 dtype 的数组，包含每个广播条目的元素算术右移。

返回类型:

数组

另请参阅

• jax.numpy.right_shift()：应用于有符号整数时，此 API 的 NumPy 包装器，也可以通过具有有符号整数 dtype 的 JAX 数组上的 x >> y 运算符访问。

• jax.lax.shift_left()：按元素的左移。

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