jax.lax.shift_right_logical

jax.lax.shift_right_logical(x, y)

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

参数:

• x (类数组)

• y (类数组)

返回类型:

数组