当前位置：网站首页>Usage of permute() function in pytorch

Usage of permute() function in pytorch

2022-08-03 03:02:00 【Mick..】

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())

原网站

版权声明
本文为[Mick..]所创，转载请带上原文链接，感谢
https://yzsam.com/2022/215/202208030140209108.html