我有一个torch.Size([1, 128, 56, 128])大小的火炬张量
1是通道，128是宽度和高度。56是图像的堆叠。
如何将其调整为torch.Size([1, 56, 128, 128])？
您可以简单地使用permute或transpose。
您可以使用swapaxes函数：
my_tensor.swapaxes(1,2)