tensorly.moveaxis

moveaxis(tensor, source, destination)

Move axes of a tensor to new positions.

Parameters
tensortl.tensor
sourceint or sequence of int

Original positions of the axes to move. These must be unique.

destinationint or sequence of int

Destination positions for each of the original axes. These must also be unique.

Returns
tensor