【Pytorch】torch. permute()

824 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第8天,点击查看活动详情

在这里插入图片描述

简介

Hello!

非常感谢您阅读海轰的文章,倘若文中有错误的地方,欢迎您指出~  

ଘ(੭ˊᵕˋ)੭

昵称:海轰

标签:程序猿|C++选手|学生

简介:因C语言结识编程,随后转入计算机专业,获得过国家奖学金,有幸在竞赛中拿过一些国奖、省奖...已保研

学习经验:扎实基础 + 多做笔记 + 多敲代码 + 多思考 + 学好英语!

唯有努力💪

简介

Hello!

非常感谢您阅读海轰的文章,倘若文中有错误的地方,欢迎您指出~  

ଘ(੭ˊᵕˋ)੭

昵称:海轰

标签:程序猿|C++选手|学生

简介:因C语言结识编程,随后转入计算机专业,获得过国家奖学金,有幸在竞赛中拿过一些国奖、省奖...已保研

学习经验:扎实基础 + 多做笔记 + 多敲代码 + 多思考 + 学好英语!

唯有努力💪

torch. permute()

语法

torch.permute(input, dims) → Tensor

  • input (Tensor) – 输入tensor
  • dims (tuple of python:ints) – 所需的维度排序

作用

返回原始张量输入的视图,其尺寸已置换。

理解为交换维度就好了

举例

x = torch.randn(2, 3, 5)
print('x.size :' , x.size())

y = x.permute(2, 0, 1) # 交换维度 第2维放在第一位,为5维,第0维在第二位,为2维....
print('y.size :', y.size())

在这里插入图片描述

注:需要注意reshape与permute的使用

参考

结语

文章仅作为个人学习笔记记录,记录从0到1的一个过程

希望对您有一点点帮助,如有错误欢迎小伙伴指正

在这里插入图片描述