Usage of permute() function in pytorch

permute() function is used for dimension transformation.It often appears in image processing. For example, the read image is in numpy format, which is (h, w, c), but the data processed by torch is (c, h, w).So this function is often used

import torch##Conform to a normal distribution with mean 0 and variance 1x=torch.randn(2,3,5)print(x.size())##Dimension transformationx=torch.permute(x,(2,0,1))print(x.size())