我有一个a= torch.randn(28, 28, 8)，我想将尺寸交换(0, 1, 2)到(2, 0, 1)。我尝试过b = a.transpose(2, 0, 1)，但收到此错误：
TypeError: transpose() received an invalid combination of arguments - got
(int, int, int), but expected one of:
* (name dim0, name dim1)
* (int dim0, int dim1)
有什么办法可以一次性全部换掉吗？