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
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.