张量结构操作

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

相关推荐
RAOY的AI笔记4 分钟前
从ChatGPT注册场景理解Web身份认证:Session、Cookie、Token与MFA基础原理
人工智能·chatgpt
vibecoding775 分钟前
AI 大模型广场选型完整指南:七大平台模型矩阵、接口兼容与定价横向对比(2026 年)
人工智能·大模型·ai编程
知几蜗牛7 分钟前
语音AI最大的误区,是默认每个人都在安静房间里说话
人工智能
A-刘晨阳14 分钟前
AI 画图到底好不好用?Next-AI-Draw.io 部署、出图与远程访问实测
人工智能·draw.io
小淮AI17 分钟前
2026年AI投研工具观察:三种技术路线的对比与思考
人工智能
学弟18 分钟前
内涵:Low-Rank Adaptation (LoRA)
人工智能
大熊背19 分钟前
研读《一种用于随机值脉冲噪声的检测统计量》中想到的一种软光敏实现算法
人工智能·软光敏·分块白平衡
YuKeeHgg31 分钟前
HuabSmart Skills:一个开源的 AI Skill、Agent、Prompt 与 MCP 能力库
人工智能·ai·prompt·华彬智融
猪脚踏浪37 分钟前
使用现有用户系统系统整合open webui
人工智能
newsxun43 分钟前
光智融合,空间新生|Jupiter SR 亮相光博会CIOE 2026
大数据·人工智能