jax.lax.shift_right_logical#

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

逐元素逻辑右移:\(x \gg y\)

此函数直接降低为 stablehlo.shift_right_logical 操作。

参数:
  • x (ArrayLike) – 输入数组。必须具有匹配的整数数据类型。如果都不是标量,xy 必须具有相同的维度数并且是广播兼容的。

  • y (ArrayLike) – 输入数组。必须具有匹配的整数数据类型。如果都不是标量,xy 必须具有相同的维度数并且是广播兼容的。

返回:

xy 具有相同数据类型的数组,其中包含每对广播条目的逐元素逻辑右移。

返回类型:

Array

另请参阅