张量结构操作

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

相关推荐
无忧智库4 分钟前
从“数据孤岛”到“城市大脑”:深度拆解某智慧城市“十五五”数字底座建设蓝图
人工智能·智慧城市
Rui_Freely6 分钟前
Vins-Fusion之 SFM准备篇(十二)
人工智能·算法·计算机视觉
hugerat9 分钟前
在AI的帮助下,用C++构造微型http server
linux·c++·人工智能·http·嵌入式·嵌入式linux
绿洲-_-15 分钟前
MBHM_DATASET_GUIDE
深度学习·机器学习
AI街潜水的八角15 分钟前
深度学习洪水分割系统2:含训练测试代码和数据集
人工智能·深度学习
万行19 分钟前
机器学习&第二章线性回归
人工智能·python·机器学习·线性回归
小宇的天下32 分钟前
HBM(高带宽内存)深度解析:先进封装视角的技术指南
网络·人工智能
rongcj37 分钟前
2026,“硅基经济”的时代正在悄然来临
人工智能
狼叔也疯狂39 分钟前
英语启蒙SSS绘本第一辑50册高清PDF可打印
人工智能·全文检索
网易易盾1 小时前
内容、行为与系统同时被利用,直播安全该如何重构
人工智能