张量结构操作

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

相关推荐
廋到被风吹走几秒前
DeepSeek V4.1-Flash-1017:一次“既要又要还要“的升级
人工智能
lll23526917791 分钟前
采访内容如何从录音变成可用资料?
人工智能
weixin_438719391 分钟前
AI 改配置:语法全对,语义全错——三道闸门照着抄
人工智能
haluhalu.6 分钟前
深入理解AIAgent(1.3)--- 设计模式
人工智能·python·设计模式·langchain
stereohomology9 分钟前
外挂可能是用AI工具自定义solomd插件和主题适合的方式
人工智能·markdown·薅羊毛·solomd·外挂新特性
东离与糖宝19 分钟前
一个句号击穿Copilot安全机制,CVSS8.8高危漏洞深度解析
人工智能
“AI国潮设计-小江”27 分钟前
《【AIGC实战】Python+SDXL批量生成“英歌舞树莓香草布丁”IP,附保持角色一致性的代码逻辑》
开发语言·人工智能·python·prompt·aigc
const_qiu28 分钟前
第四章 06:4.6 统一入口 —— analyze() 能力收束
java·人工智能
有为少年36 分钟前
容器内存显示 100%、进程莫名被 Killed 解决方案
机器学习·容器·模型训练·模型推理
硅谷秋水41 分钟前
DynaHarness 方法总结
人工智能·机器学习·语言模型·机器人