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