PyTorch中permute的用法

permute(dims)函數 將tensor的維度換位。.net 參數：參數是一系列的整數，表明原來張量的維度。好比三維就有0，1，2這些dimension。blog 例：token import torch import numpy    as np圖片 a=np.array([[[1,2,3],[4,5,6]]])get unpermuted=torch.tensor(a) print(un

>>阅读原文<<