mars.tensor.flip

mars.tensor.flip(m, axis)

Reverse the order of elements in a tensor along the given axis.

The shape of the array is preserved, but the elements are reordered.

参数

• m (array_like) – Input tensor.

• axis (integer) – Axis in tensor, which entries are reversed.

返回

out – A view of m with the entries of axis reversed. Since a view is returned, this operation is done in constant time.

返回类型

array_like

参见

flipud

Flip a tensor vertically (axis=0).

fliplr

Flip a tensor horizontally (axis=1).

提示

flip(m, 0) is equivalent to flipud(m). flip(m, 1) is equivalent to fliplr(m). flip(m, n) corresponds to m[...,::-1,...] with ::-1 at position 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

上一页

mars.tensor.unique

下一页

mars.tensor.fliplr