mindspore.numpy.rollaxis
- mindspore.numpy.rollaxis(x, axis, start=0)[source]
Rolls the specified axis backwards, until it lies in the given position. The positions of the other axes do not change relative to one another.
- Parameters
x (Tensor) – A Tensor to be transposed.
axis (int) – The axis to be rolled.
start (int) –
Default: 0. If \(start <= axis\), the axis is rolled back until it lies in this position (start). If \(start > axis\): the axis is rolled until it lies before this position (start). If \(start < 0\), the start will be normalized as a non-negative number (more details can be seen in the source code.)
- Returns
Transposed Tensor. Has the same data type as the original tensor x.
- Supported Platforms:
Ascend
GPU
CPU
- Raises
TypeError – If axis or start is not integer, or x is not tensor.
ValueError – If axis is not in the range of \([-ndim, ndim-1]\) or start is not in the range of \([-ndim, ndim]\).
Examples
>>> import mindspore.numpy as np >>> x = np.ones((2,3,4)) >>> output = np.rollaxis(x, 0, 2) >>> print(output.shape) (3, 2, 4)