mlx.core.roll

roll(a: array, shift: int | Tuple[int], axis: None | int | Tuple[int] = None, /, *, stream: None | Stream | Device = None) → array

沿给定轴滚动数组元素。

滚动超出数组末尾的元素会从开头引入，反之亦然。

如果未提供轴，则数组会被展平、滚动，然后恢复形状。

参数:

• a (array) – 输入数组

• shift (int 或 tuple(int)) – 元素移位的位数。如果为正，则数组向右滚动；如果为负，则向左滚动。如果提供的是 int 而 axis 是一个 tuple，则所有轴都使用相同的值。

• axis (int 或 tuple(int)，可选) – 沿其滚动元素的轴。