tensorly.moveaxis

moveaxis(tensor, source, destination)

Move axes of a tensor to new positions.

Parameters:

tensor : tl.tensor

source : int or sequence of int

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

destination : int or sequence of int

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

Returns:

tensor