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