张量结构操作

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

相关推荐
zuozewei23 分钟前
零基础 | AI应用记忆管理:从短期到长期的完整实践指南
运维·服务器·人工智能
数说星榆18124 分钟前
小型工厂工艺流程图制作_在线设计装配/焊接/冲压工艺流程模板
大数据·论文阅读·人工智能·流程图·论文笔记
老蒋每日coding31 分钟前
AI Agent 设计模式系列(十九)—— 评估和监控模式
人工智能·设计模式
AI浩34 分钟前
用于自动驾驶的ApolloScape数据集
人工智能·机器学习·自动驾驶
weixin_4215850140 分钟前
无监督配准
人工智能
救救孩子把42 分钟前
56-机器学习与大模型开发数学教程-5-3 最速下降法与动量法(Momentum)
人工智能·机器学习
njsgcs1 小时前
MiniCPM4-0.5B-QAT-Int4-GPTQ-format 小显存llm
linux·人工智能
koo3641 小时前
pytorch深度学习笔记18
pytorch·笔记·深度学习
Paul-LangJun1 小时前
互信息和InfoNCE的关系
人工智能·对比学习·infonce·互信息
lixin5565561 小时前
基于神经网络的音乐生成增强器
java·人工智能·pytorch·python·深度学习·语言模型