pytorch 调整tensor的维度位置

target.permute([0, 3, 1, 2])

 

一定要使用permute以及中括号

 

一些在我这里没起到作用的网上的例子：

1. https://blog.csdn.net/zouxiaolv/article/details/80936725

preds = to_numpy(preds)#preds是[2985x16x2]
preds = preds.transpose(2, 1, 0)#preds[2x16x2985]

 

2. https://www.jb51.net/article/152793.htm

import torch
 
x = torch.linspace(1, 9, steps=9).view(3, 3)
 
b=x.permute(1,0)
print(b)
print(b.permute(1,0))