mars.tensor.roll#

mars.tensor.roll(a, shift, axis=None)[来源]#

沿给定轴滚动张量元素。

超出最后位置的元素将重新引入到第一个位置。

Parameters

a (array_like) – 输入张量。
shift (int 或 tuple 的 ints) – 元素移动的位数。如果是一个元组，那么 axis 必须是相同大小的元组，每个给定的轴都按相应的数量移动。如果是一个整数而 axis 是一个整数元组，则相同的值用于所有给定的轴。
axis (int 或 tuple 的 ints, 可选) – 元素被移动的轴或维度。默认情况下，张量在移动之前会被展平，之后会恢复原始形状。

Returns

res – 输出张量，形状与 a 相同。

Return type

张量

另请参阅

rollaxis: 将指定的轴向后滚动，直到它位于给定的位置。

备注

支持同时在多个维度上滚动。

示例

>>> import mars.tensor as mt

>>> x = mt.arange(10)
>>> mt.roll(x, 2).execute()
array([8, 9, 0, 1, 2, 3, 4, 5, 6, 7])

>>> x2 = mt.reshape(x, (2,5))
>>> x2.execute()
array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])
>>> mt.roll(x2, 1).execute()
array([[9, 0, 1, 2, 3],
       [4, 5, 6, 7, 8]])
>>> mt.roll(x2, 1, axis=0).execute()
array([[5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4]])
>>> mt.roll(x2, 1, axis=1).execute()
array([[4, 0, 1, 2, 3],
       [9, 5, 6, 7, 8]])