张量结构操作

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

相关推荐
罗小罗同学2 分钟前
Nat. Biomed. Eng最新发表的医学AI基础模型CLEAR,可以将诊断决策与临床概念一一匹配,不再是传统黑箱模型
人工智能·医学图像处理·医工交叉·医学ai
Leslie1652 分钟前
多人共享目录如何不失控:用 setgid、ACL、umask 与粘滞位设计 Linux 权限
人工智能
mit6.8247 分钟前
“Copilot“ 超级应用“年内问世(`・ω・´)
人工智能
zhbcddxr11 分钟前
GEO服务商度量监测能力测评:可见度追踪与报表排行
大数据·人工智能·microsoft
m沐沐12 分钟前
【自然语言处理】词向量转换与中文文本情感分类——从CountVectorizer到朴素贝叶斯
人工智能·算法·机器学习·自然语言处理·分类·中文分词·词向量转换
科技发布26 分钟前
拆解传播易服务模式,看清小红书团购从入驻变现完整逻辑
大数据·人工智能
用户2986985301431 分钟前
Excel 转 PDF 免费在线工具,格式完美保留
人工智能·c#·excel
武子康38 分钟前
2026 年 7 月 AI 模型发布复盘:真正被比较的是整套工作系统
人工智能·llm·agent
qq_4195630938 分钟前
1.8-从深度学习到大模型时代
人工智能·深度学习
miaowu35740 分钟前
企业 AI 智能体实施周期短吗?各阶段费用明细
人工智能