Pytorch tensor to change dimension
I have a RGB image tensor as (3,H,W), but the plt.imshow() can not show RGB image with this  shape. I want to change the tensor to (H,W,3). How can I do that, is pytorch function .view() can do that?

An alternative to using torch.Tensor.permute is to apply torch.Tensor.movedim:
image.movedim(0,-1)

Which is actually more general than image.permute(1,2,0), since it works for any number of dimensions. It has the effect of moving axis=0 to axis=-1 in a sort of insertion operation.
Or equivalently with Numpy, using np.moveaxis:
