mxnet.np.rollaxis

rollaxis(a, axis, start=0)

Roll the specified axis backwards, until it lies in a given position.

Parameters

  • a (ndarray) – Input array.

  • axis (integer) – The axis to roll backwards. The positions of the other axes do not change relative to one another.

  • start (int, optional) – The axis is rolled until it lies before this position. The default, 0, results in a “complete” roll.

Returns

  • res (ndarray) – A view after applying rollaxis to a is returned.

  • —–

Examples

>>> a = np.ones((3,4,5,6))
>>> np.rollaxis(a, 3, 1).shape
(3, 6, 4, 5)
>>> np.rollaxis(a, 2).shape
(5, 3, 4, 6)
>>> np.rollaxis(a, 1, 4).shape
(3, 5, 6, 4)