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
提示
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