PyTorch中Tensor.permute作用

解释

将 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]的维度。