Pytorch Tensor维度变换方法

1.torch.reshape()、torch.view()可以调整Tensor的shape

2.torch.unsqueeze(index)可以为Tensor增加一个维度

3.squeeze()可以删减维度

4.expand()扩展维度

5.repeat()维度重复,不常用

6 .transpose(dim1, dim2)交换dim1与dim2,进行维度变换。

注意这种交换使得存储不再连续,再执行一些reshape的操作会报错,所以要调用一下contiguous()使其变成连续的维度。

参考:https://blog.csdn.net/weicao1990/article/details/93618136

7.Broadcasting(广播)也就是自动实现了若干unsqueeze和expand操作,以使两个Tensor的shape一致,从而完成某些操作(往往是加法)。

比如得到的feature maps的维度是4,32,14,14,对这些特征图加上偏置,bias32,先对bias进行unsqueeze操作增加维度:1,32,1,1,再用expand操作进行扩充4,32,14,14。然后两者就可以相加了。

相关推荐
SEO_juper3 分钟前
2026_GEO_AI搜索技术实战_CSDN
人工智能·chrome·目标检测·seo·geo·谷歌优化
暴躁的大熊5 分钟前
机器学习中的知识图谱:概念、核心技术与应用场景
人工智能·机器学习·知识图谱
Pokerhead6 分钟前
如何评价 DeepSeek-V4 的价格?
人工智能·大模型·ai编程·deepseek
aiqianji9 分钟前
有哪些稳定的AI短篇小说写作软件可以推荐?
人工智能·python
蓝速科技10 分钟前
蓝速科技丨AI双屏翻译机:重塑线下接待的从容沟通体验
人工智能·科技
Nomarsgo11 分钟前
研华17寸工业显示器 FPM-217在智能制造产线中的应用方案
人工智能·科技·计算机视觉·视觉检测·电脑
ifenxi爱分析12 分钟前
爱分析发布2026年中国GEO市场研究报告
人工智能
石小石Orz17 分钟前
TRAE SOLO实战:实现一个桌面3D助手
前端·人工智能
武子康27 分钟前
Ask/Allow 不是安全边界:企业 Coding Agent 必须建立四层治理(Policy / Scoped Credential / Sandbox / Provenance)
人工智能·后端·agent
独隅28 分钟前
DevEco Code 在 Windows/MacOS 双系统上的完整使用指南
ide·人工智能·windows·macos·华为·harmonyos