其他分享
首页 > 其他分享> > pytorch 中permute用法

pytorch 中permute用法

作者:互联网

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

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

 

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

 

标签:四维张量,不同,张量,用法,pytorch,维度,parameter,permute
来源: https://www.cnblogs.com/-taole/p/15314713.html