How to reshape torch.Size([2, 3, 4, 5]) to torch.Size([2, 5, 3, 4]) and then get back to torch.Size([2, 3, 4, 5])?
In my case torch.Size([batch_size, x_dim, y_dim, 3_RGB]). I need convert to torch.Size([batch_size, 3_RGB, x_dim, y_dim]) and use it with CNN layer and then get back to original size: torch.Size([batch_size, x_dim, y_dim, 3_RGB])