jax.lax.shift_right_logical#
- jax.lax.shift_right_logical(x, y)[source]#
逐元素逻辑右移:\(x \gg y\)。
此函数直接降低为 stablehlo.shift_right_logical 操作。
- 参数:
x (ArrayLike) – 输入数组。必须具有匹配的整数数据类型。如果都不是标量,
x
和y
必须具有相同的维度数并且是广播兼容的。y (ArrayLike) – 输入数组。必须具有匹配的整数数据类型。如果都不是标量,
x
和y
必须具有相同的维度数并且是广播兼容的。
- 返回:
与
x
和y
具有相同数据类型的数组,其中包含每对广播条目的逐元素逻辑右移。- 返回类型:
另请参阅
jax.numpy.right_shift()
:应用于无符号整数时此 API 的 NumPy 包装器,也可通过 JAX 数组上带有无符号整数数据类型的x >> y
运算符访问。jax.lax.shift_left()
:逐元素左移。jax.lax.shift_right_arithmetic()
:逐元素算术右移。