张量结构操作

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

相关推荐
witAI14 小时前
**AI漫剧制作工具2025推荐,解锁高效创作新体验**
人工智能·python
qinyia14 小时前
如何在服务器上查看网络连接数并进行综合分析
linux·运维·服务器·开发语言·人工智能·php
新缸中之脑14 小时前
构建一个论文学习AI助手
人工智能·学习
说私域14 小时前
私域流量生态重构:链动2+1模式S2B2C商城小程序的流量整合与价值创造
人工智能·小程序·流量运营·私域运营
圆奋奋14 小时前
让“小爱音箱PRO”智能起来:接入豆包AI
人工智能
aiguangyuan14 小时前
使用PyTorch和Hugging Face Transformers构建GPT教学演示:从基础原理到实践应用
人工智能·python·nlp
自可乐14 小时前
Apache Airflow完全学习指南:从入门到精通的系统教程
人工智能·机器学习·apache
说私域14 小时前
AI智能名片S2B2C商城小程序赋能下线上向线下导流的机制与效果研究——基于线下专属优惠券的视角
人工智能·小程序·流量运营·私域运营
朴实赋能14 小时前
2026跨境电商生死局:AI大模型重构购物链路,智矩引擎打造品牌出海“自动驾驶”系统
人工智能·社媒矩阵·文旅出海·海外社媒引流·ai大模型跨境电商·shopify独立站引流·社媒矩阵流量创造
啊巴矲14 小时前
小白从零开始勇闯人工智能:计算机视觉初级篇(OpenCV补充(1))
人工智能·opencv·计算机视觉