mars.tensor.flip#

mars.tensor.flip(m, axis)[来源]#

沿给定轴反转张量中元素的顺序。

数组的形状被保留，但元素的顺序被重新排列。

Parameters

  • m (array_like) – 输入张量。

  • axis (integer) – 张量中的轴，其条目被反转。

Returns

out – 一个 m 的视图，其轴的条目被反转。由于返回的是视图，因此此操作在恒定时间内完成。

Return type

类数组对象

另请参阅

flipud

垂直翻转一个张量（轴=0）。

fliplr

水平翻转一个张量 (轴=1)。

备注

flip(m, 0) 等价于 flipud(m)。 flip(m, 1) 等价于 fliplr(m)。 flip(m, n) 对应于 m[...,::-1,...]，其中 ::-1 在位置 n。

示例

>>> import mars.tensor as mt

>>> A = mt.arange(8).reshape((2,2,2))
>>> A.execute()
array([[[0, 1],
        [2, 3]],
[[4, 5],

[6, 7]]])

>>> mt.flip(A, 0).execute()
array([[[4, 5],
        [6, 7]],
[[0, 1],

[2, 3]]])

>>> mt.flip(A, 1).execute()
array([[[2, 3],
        [0, 1]],
[[6, 7],

[4, 5]]])

>>> A = mt.random.randn(3,4,5)
>>> mt.all(mt.flip(A,2) == A[:,:,::-1,...]).execute()
True