pytorch 中permute用法

permute(parameter): permute中参数为张量的维度,将不同维度以不同的维度排列,得到一个新的张量,在深度学习中的主要作用是将特征值聚类。具体使用如下:

1. 首先定义一个四维张量

pytorch 中permute用法

 

 2. 使用permute函数将不同维度顺序转换

pytorch 中permute用法

 

上一篇:Java || 看了大二学长写的代码,我竟开始默默的模仿了。。。


下一篇:MySQL:Parameter index out of range (5 > number of parameters, which is 1)..解决