NJU1healer的个人博客分享 http://blog.sciencenet.cn/u/NJU1healer

博文

PyTorch中permute的用法

已有 404 次阅读 2020-9-25 10:16 |个人分类:Pytorch|系统分类:科研笔记

permute(dims)

将tensor的维度换位。

参数:参数是一系列的整数,代表原来张量的维度。比如三维就有0,1,2这些dimension。

例:

import torch

import numpy    as np


a=np.array([[[1,2,3],[4,5,6]]])


unpermuted=torch.tensor(a)

print(unpermuted.size())  #  ——>  torch.Size([1, 2, 3])


permuted=unpermuted.permute(2,0,1)

print(permuted.size())     #  ——>  torch.Size([3, 1, 2])


        再比如图片img的size比如是(28,28,3)就可以利用img.permute(2,0,1)得到一个size为(3,28,28)的tensor。


利用这个函数permute(0,2,1)可以把Tensor([[[1,2,3],[4,5,6]]]) 转换成

tensor([[[1., 4.],

        [2., 5.],

        [3., 6.]]])


如果使用view,可以得到

tensor([[[1., 2.],

         [3., 4.],

         [5., 6.]]])


关于view的用法:参见PyTorch中view的用法 

【参考】

https://blog.csdn.net/york1996/article/details/81876886

点滴分享,福泽你我!Add oil!



http://blog.sciencenet.cn/blog-3428464-1252015.html

上一篇:Pytorch中的Squeeze()和Unsqueeze()函数
下一篇:model.train()与model.eval()的用法

0

该博文允许注册用户评论 请点击登录 评论 (0 个评论)

数据加载中...

Archiver|手机版|科学网 ( 京ICP备07017567号-12 )

GMT+8, 2020-10-29 23:08

Powered by ScienceNet.cn

Copyright © 2007- 中国科学报社

返回顶部