如何交换火炬张量中的轴？

Question

我有一个大小为的火炬张量torch.Size([1, 128, 56, 128])

1是通道，128是宽度和高度。56 是图像堆栈。

我怎样才能将其调整为torch.Size([1, 56, 128, 128])？

Answer 1

您可以简单地使用permute或transpose。