target.permute([0, 3, 1, 2])
一定要使用permute以及中括号一定要使用
阅读背景：
pytorch 调整tensor的维度位置
来源：互联网
分享到：
