Pytorch之permute函数

1、主要作用：变换tensor维度 example： import torch x = torch.randn(2, 3, 5) print(x.size()) print(x.permute(2, 0, 1).size()) >>>torch.Size([2, 3, 5]) >>>torch.Si
posted @ 2020-03-10 23:01  嶙羽  阅读(4952)  评论(0)  编辑  收藏  举报