permute(parameter): permute中参数为张量的维度,将不同维度以不同的维度排列,得到一个新的张量,在深度学习中的主要作用是将特征值聚类。具体使用如下:
1. 首先定义一个四维张量
2. 使用permute函数将不同维度顺序转换
2024-02-15 17:08:40
permute(parameter): permute中参数为张量的维度,将不同维度以不同的维度排列,得到一个新的张量,在深度学习中的主要作用是将特征值聚类。具体使用如下:
1. 首先定义一个四维张量
2. 使用permute函数将不同维度顺序转换