双方.flatten()并.view(-1)压平PyTorch张量。有什么不同?
.flatten()
.view(-1)
python pytorch
我有一个大小为的火炬张量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
python ×2
pytorch ×2
permutation ×1