网站首页
编程语言
数据库
系统相关
其他分享
编程问答
0.4145
2024-10-24
pytorch中的permute()函数的原理
permute的中文含义是:置换这个就已经解释了一半了。他的功能就是把一个多维张量的维度进行调换。对于一个2维张量来说,例如:a=torch.rand(2,3).那么代码a.permute(1,0)就相当于a.T对于一个三维张量来说,例如:a=torch.rand(2,3,4)代码a.permute(2,0,1)把最初的2,3,4维度,给调整成了