我有一个大小为的火炬张量torch.Size([1, 128, 56, 128])
torch.Size([1, 128, 56, 128])
1是通道，128是宽度和高度。56 是图像堆栈。
我怎样才能将其调整为torch.Size([1, 56, 128, 128])？
torch.Size([1, 56, 128, 128])
python permutation pytorch
permutation ×1
python ×1
pytorch ×1