解释
将 Tensor 的维度换位置。
例子
我们发现,第一维和第二维的维数互换了,permute 常用于训练验证数据时维度位置不同的情况,比如训练时使用[B, C, H, W]而当前张量是[H, W, C]的话,就要先用 unsqueeze(0)加一个维度成[B, H, W, C],再 permute(0, 3, 1, 2) 变换成[B, C, H, W],要思考清楚是从哪种变换成哪种,目标类型是[B, C, H, W],而 permute 的参数对应的是源类型[B, H, W, C]的维度。