张量结构操作

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

相关推荐
长谷深风11110 分钟前
根因定位:三步隔离法破解AIBadcase
人工智能·ai·大模型·retrieval·ai智能体·aiagent·aibadcase
陈碧甫14 分钟前
元龙因果链的因果生成式写作
人工智能
deepseek2314 分钟前
GPT-6 Astra 破解 FrontierMath 九年悬案拆解:调和熵投票规则反证核恒非空,局部搜索如何终结反例悬赏
人工智能·算法·ai agent
Chukai12319 分钟前
AI智能体:会思考会干活的下一代AI
人工智能·agent
9呀21 分钟前
[skill] agent 自动保存聊天记录到md文件 skill auto-save-md
人工智能
怕浪猫21 分钟前
一行命令复刻爆款视频,我把 Hypit 从安装跑到了出片
人工智能·设计模式·程序员
Raas10028 分钟前
MAI Gateway(魔芋企业级AI网关)功能全解:AI网关支持本地模型吗?一文看懂AI网关能力矩阵
大数据·人工智能·网关·ai网关·mai gateway·企业级产品
蜗牛互联网32 分钟前
AI给面试打分不够,求职者更需要可核对的证据
java·人工智能·后端
不开大的凯207734 分钟前
跑分已死,交付为王
人工智能·ai·麦当秀aippt·ai office
AI多Agent协作实战派40 分钟前
【AI探索历程19】飞牛版:给NAS用户的一键部署
人工智能