cupy.roll

cupy.roll(a, shift, axis=None)

沿给定轴滚动数组元素。

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

参数:

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

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

• axis (int or tuple of int or None) – 元素沿其移动的轴。默认情况下，数组在移动前会被展平，移动后会恢复原始形状。

返回:

输出数组。

返回类型:

ndarray

参见

numpy.roll()