Pytorch Tensor维度调换

permute

如果四个维度表示上节的[batch,channel,h,w] ，如果想把channel放到最后去，形成[batch,h,w,channel]，那么如果使用前面的维度交换，至少要交换两次（先13交换再12交换）。而使用permute可以直接指定维度新的所处位置，更加方便。

示例代码：

a = torch.rand(4, 3, 6, 7)
print(a.permute(0, 2, 3, 1).shape)
输出结果：

torch.Size([4, 6, 7, 3])
————————————————
版权声明：本文为CSDN博主「洪流之源」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weicao1990/article/details/93618136