张量结构操作

torch.transpose可以交换张量的维度,torch.transpose常用于图片存储格式的变换上。

#在第0维插入长度为1的一个维度

d = torch.unsqueeze(s,axis=0)

改成 (3,6)形状的张量

b = a.view(3,6) #torch.reshape(a,3,6)

print(b.shape)

print(b)

改回成 1,3,3,2 形状的张量

c = torch.reshape(b,1,3,3,2) # b.view(1,3,3,2)

print(c)

如果张量在某个维度上只有一个元素,利用torch.squeeze可以消除这个维度。

torch.unsqueeze的作用和torch.squeeze的作用相反。

可以用torch.cat方法和torch.stack方法将多个张量合并,可以用torch.split方法把一个张量分割成多个张量。

torch.cat和torch.stack有略微的区别,torch.cat是连接,不会增加维度,而torch.stack是堆叠,会增加维度。 (这个是把多张图片张量合并成一个batch用的张量操作方法)

相关推荐
MetaLite2 分钟前
AI编程与工程底座-让AI遵守SpringBoot边界
人工智能·spring boot·ai编程
angered8 分钟前
「AI 应用 / AI Agent」行业日报 · 2026-09-05
人工智能·ai编程
CAIE注册人工智能工程师9 分钟前
突发!GPT-6刚刚开放使用、上线API,能体验AGI模型了
人工智能·gpt·agi
weixin_4402132910 分钟前
7大机器学习范式:有监督/自监督/半监督/主动学习/弱监督/自训练/迁移学习
机器学习·迁移学习·自监督学习·有监督学习·主动学习
进击切图仔17 分钟前
Autodl 平台接受客户端端口数据
人工智能
新知图书18 分钟前
第8章 智能体设计模式5:上下文工程
人工智能·智能体
IT_陈寒22 分钟前
Python的多线程居然是个假把式?搞清GIL让我少熬三天夜
前端·人工智能·后端
熊野君34 分钟前
第 7 章 AI时代产品经理新增能力
大数据·人工智能·经验分享·职场和发展·产品经理
AcaDesign38 分钟前
省部级协会科学技术奖答辩ppt视频制作_科技进步奖_技术发明奖_自然科学奖PPT模板
人工智能
思考着亮1 小时前
7.限流、安全防护、缓存分流与模型容灾
人工智能