pytorch中的Permute转换
permute(dims)
将tensor的维度换位。
参数：参数是一系列的整数，代表原来张量的维度。比如三维就有0，1，2这些dimension。
举个例子
import torch
import numpy as npa=np.array([[[1,2,3,4],[4,5,6,7]]])unpermuted=torch.tensor(a)
print(a)
print(unpermuted.size()) # ——> torch.Size([1, 2, 3])permuted=unpermuted.permute(2,0,1)
print(permuted.size()) # ——> torch.Size([3, 1, 2])print(permuted)
参考资料
PyTorch中permute的用法_York1996的博客-CSDN博客_permute pytorch
Pytorch view() permute() contiguous() transpose()_我是天才很好的博客-CSDN博客
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！