cupy.roll#

cupy.roll(a, shift, axis=None)[source]#

沿给定轴滚动数组元素。

滚动超出最后一个位置的元素会重新出现在第一个位置。

参数:
  • a (ndarray) – 要滚动的数组。

  • shift (intint 元组) – 元素移动的位数。如果是一个元组,则 axis 必须是相同大小的元组,并且每个给定轴都按相应的位数移动。如果是一个整数而 axis 是一个整数元组,则所有给定轴都使用相同的值。

  • axis (intint 元组None) – 元素沿其移动的轴。默认情况下,数组在移动前被展平,之后恢复原始形状。

返回值:

输出数组。

返回类型:

ndarray

另请参阅

numpy.roll()