pytorch之permute()用法

将tensor的维度换位。

参数： - dims (int …*) - 换位顺序

例：

>>> x = torch.randn(2, 3, 5) 
>>> x.size() 
torch.Size([2, 3, 5]) 
>>> x.permute(2, 0, 1).size() 
torch.Size([5, 2, 3])

Reference

1. Pytorch之permute函数