目录
- 简介
- torch. permute()
- 语法
- 作用
- 举例
- 参考
- 结语
简介
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的使用
参考
- https://pytorch.org/docs/stable/generated/torch.permute.html#torch.permute
结语
文章仅作为个人学习笔记记录，记录从0到1的一个过程
希望对您有一点点帮助，如有错误欢迎小伙伴指正
