请选择 进入手机版 | 继续访问电脑版
查看: 11|回复: 1

PyTorch中permute的用法

[复制链接]

156

主题

599

帖子

6292

积分

xdtech

Rank: 5Rank: 5

积分
6292
发表于 2019-12-2 21:16:33 | 显示全部楼层 |阅读模式
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.]]])


回复

使用道具 举报

156

主题

599

帖子

6292

积分

xdtech

Rank: 5Rank: 5

积分
6292
 楼主| 发表于 2019-12-2 21:16:38 | 显示全部楼层
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表