张量结构操作

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

相关推荐
Kel2 分钟前
万物皆 Runnable:LangChain 如何用一个接口统一模型、链与图
人工智能·设计模式·架构
科技圈快迅2 分钟前
解决续费痛点,轻量化AI订阅替代方案深度解读
大数据·人工智能
Tbisnic5 分钟前
23.大模型开发:深度学习----CNN 卷积神经网络 与 RNN 循环神经网络
人工智能·python·rnn·深度学习·cnn·ai大模型
灵机一物12 分钟前
2026算力推荐进入专业化阶段:超互联品牌如何构建企业AI选型能力矩阵
服务器·人工智能·决策树·云平台
2zcode19 分钟前
基于MATLAB语音信号分析与机器学习的帕金森病智能检测系统
人工智能·机器学习·matlab
冬奇Lab22 分钟前
MCP Server 入门开发 + 协议调试 + 生产级部署
人工智能·llm·mcp
冬奇Lab22 分钟前
开源项目第157期:archify — 用自然语言生成可主题切换、高分辨率导出的架构图
人工智能·架构·claude
2301_8002561124 分钟前
人工智能入门学习——ANN、CNN基本概念和考试核心代码
人工智能·学习·cnn
A hao25 分钟前
HDR LED显示屏工作原理:高动态范围显示技术解析
大数据·图像处理·人工智能
2601_9549711328 分钟前
二本应用统计学专业就业前景与岗位分析:CDA数据分析师证书的价值
深度学习