torch.js has not been released yet.

torch.movedim

function movedim(input: Tensor, source: number | number[], destination: number | number[]): void

Moves one or more dimensions of the input tensor to new positions.

Parameters

inputTensor: The input tensor
sourcenumber | number[]: The source dimension(s) to move
destinationnumber | number[]: The destination dimension(s) for each source

Returns

Tensor with dimensions moved to new positions

Examples

const x = torch.randn(2, 3, 4, 5);
torch.movedim(x, 0, -1).shape;  // [3, 4, 5, 2] - move dim 0 to end
torch.movedim(x, [0, 1], [2, 3]).shape;  // [4, 5, 2, 3]

See Also

PyTorch torch.movedim()