张量结构操作

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用的张量操作方法)

相关推荐
Forerror20261 分钟前
深入理解大模型网关解决什么问题:MAIGateway(魔芋企业级AI网关)架构与核心价值解读
人工智能·ai网关·maigateway·大模型财务管控·企业级大模型治理网关
AI深栈4 分钟前
第 10 章 · Embedding、VectorStore 与 RAG
java·人工智能
jsl_jsl_jsl7 分钟前
《一个 Agent 平台怎么接入多家大模型:Provider 槽位制设计》
人工智能
ι:11 分钟前
Codex 自主调用 Visio 绘图完整教程
人工智能·visio·codex
梦帮科技11 分钟前
一次性会员支付系统的可靠性设计:幂等、金额校验、Webhook 与链上确认
人工智能·神经网络·mysql·区块链·建造者模式·合成复用原则·加密货币
维克兜率天13 分钟前
【维克】模块3总结:从一行空数据,到一个能跑的模型
python·深度学习·算法
AgentMaster14 分钟前
智能客服系统技术选型实战:从架构设计到落地实施的完整指南
大数据·人工智能
lailai041016 分钟前
图像处理的技术路径与实现方式考察
人工智能
回眸&啤酒鸭22 分钟前
【回眸】GPT-5.6 Luna 批量处理实战指南
人工智能
suaizai_26 分钟前
多智能体架构揭秘:从混乱到高效
人工智能