Pytorch的tensor操作 permute()函数 permute()函数 是对tensor的维度位置进行变换 具体理解看代码 a = torch.rand(1, 2, 3) print(a.shape) a = a.permute(2, 1, 0) print(a.shape) torch.Size([1, 2, 3]) torch.Size([3, 2, 1])