张量结构操作

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

相关推荐
MacroZheng几秒前
面试官皱眉:"你懂 Vibe Coding,那你说superpowers和grill-me怎么选?",我:"小孩才做选择,我全都要!"
java·人工智能·后端
OceanBase数据库官方博客1 分钟前
深度拆解seekdb:AI Native Database 的技术架构与核心能力
数据库·人工智能·架构
大模型码小白2 分钟前
【AI大模型】DeepSeek Harness 深度解析:大模型评测框架的架构与实践
java·运维·人工智能·spring·架构·自动化
码农小旋风3 分钟前
GPT-6 Astra 直接进 Blender,AI 开始从会说变成会做
人工智能·gpt·blender
武子康3 分钟前
9/10 和 900/1000 都是 90%:机器人成功率计算器必须阻止的误判
人工智能·llm·agent
weixin_618232305 分钟前
OpenAI智能体“劫持”德国老网站当留言板
人工智能·安全
未来之窗软件服务5 分钟前
城市街道社区综合管理开发之档案管理—东方仙盟
人工智能·仙盟创梦ide·档案系统
武哥聊编程5 分钟前
【AI实战项目】基于SpringAI+Springboot+Vue的AI智能简历优化助手
vue.js·人工智能·spring boot
phoenix-bai5 分钟前
表格神经网络架构发展史:从 MLP 到表格基础模型的二十年
人工智能·神经网络·架构
大虾别跑10 分钟前
ai-daily-2026-09-07
人工智能