张量结构操作

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

相关推荐
hguisu4 分钟前
AI大模型-2:智能体(Agent)和demo演示实践
人工智能
墨染天姬6 分钟前
【AI】英伟达 AVO 自动生成GPU算子
人工智能
向量引擎10 分钟前
肝了三天三夜!四大AI模型(DeepSeek/Gemini/ChatGPT/豆包)深度横评,开发者该如何选?
人工智能·chatgpt·架构·开源·aigc·文心一言·api调用
liliangcsdn19 分钟前
如何使用向量库faiss和LLM判断问题是否被记录
人工智能·全文检索·faiss
m0_6515939129 分钟前
从“只会鹦鹉学舌”到“能独当一面”:以人的成长为例,看懂大模型的成长史与未来
人工智能
阳艳讲ai30 分钟前
企业AI培训评测:线下课vs线上营vs陪跑服务,谁能让员工真学会AI?附ROI选型指南
大数据·人工智能·数据挖掘
人道领域31 分钟前
注意!2026大模型风口突变:告别云端军备竞赛,端侧AI才是今年最大黑马
人工智能·大模型·端侧ai
YakSue32 分钟前
让AI帮我整理GDC游戏开发资源
人工智能
Dev7z33 分钟前
基于YOLOv8面向家居场景的火焰烟雾图像识别系统
人工智能·yolo
hughnz36 分钟前
油气运营商加速采用 AI 驱动软件
人工智能