Reshape pytorch tensor

i have Video Tensor Size: torch.Size([15, 3, 112, 112])
i want to reshape all tensor to [3,15,112,112] it is possible ?

If the variable is x then x.permute(1,0,2,3) might work in your case.

1 Like