mlx.core.right_shift

right_shift(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array

逐元素的右移操作。

使用 numpy 风格的广播语义，将第一个输入的位向右移动第二个输入的位数。两个输入数组都可以是标量，或者一个是数组一个是标量。

参数:

• a (array) – 输入数组或标量。

• b (array) – 输入数组或标量。

返回值:

按位右移的结果 a >> b。

返回类型:

array