(PyTorch) permute(dims)方法
方法说明
permute是pytorch中tensor的方法
用于调换tensor的维度。
例子
>>> x = torch.randn(2, 3, 5)
>>> x.size()
torch.Size([2, 3, 5])
>>> x.permute(2, 0, 1).size()
torch.Size([5, 2, 3])
原始tensor的第0, 1, 2这三个维度分别为2,3,5。permute方法重新指定维度。使用permute之后的的第0,1,2三个维度分别为5,2,3.