张量结构操作

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

相关推荐
statistican_ABin2 分钟前
中国城市 PM2.5较高浓度风险识别
人工智能
茶马古道的搬运工3 分钟前
AI 深度技能之-解读Multica(一)- 让Agent 变成真·队友
人工智能
YOLO数据集集合4 分钟前
自动驾驶道路视觉开源数据集TT100K工程化使用文档|道路标识目标检测资源|开源数据集分享
人工智能·目标检测·自动驾驶
BerryS3N6 分钟前
Cursor+GitOps:AI驱动的自动化运维新范式
运维·人工智能·自动化
力学与人工智能11 分钟前
POF | 西工大曹文博、张伟伟等:基于时间步进导向的神经网络求解高雷诺数壁湍流问题
人工智能·神经网络·机器学习·高雷诺数·湍流
吾鳴12 分钟前
GPT-5.6实测,OpenAI太良心了!
人工智能
ComputerInBook21 分钟前
opencv 图像处理之边缘检测
图像处理·人工智能·opencv·边缘检测
卡梅德生物科技小能手21 分钟前
卡美德生物科普NGF(神经生长因子):神经系统发育与修复的关键调节因子
经验分享·深度学习·生活
大鱼>25 分钟前
宠物异常行为预警系统:边缘计算与实时检测
人工智能·深度学习·算法·iot·宠物
萧青山26 分钟前
AI阅读增强套件:用苏格拉底诘问+对抗性阅读+知识图谱构建深度阅读技能套件(Python实现)
人工智能·python·知识图谱·ai阅读增强