首页
归档
分类
标签
动态
图库
藏宝阁
NeoDB
120s视界
工具
IT-Tools
白板工具
关于我
穿梭
虫洞
异次元之旅
1
记录一次汗流浃背的网站回滚
2
使用Brevo配置邮件通知
3
Alist + Duplicati 实现Halo云备份
4
docker设置网络代理
5
2023年度总结
JacyLunatic
Total posts:
83
Total categories:
15
Total tags:
10
Navigation
首页
归档
分类
标签
动态
图库
藏宝阁
NeoDB
120s视界
工具
IT-Tools
白板工具
关于我
穿梭
虫洞
异次元之旅
Table of Contents
Tag
Pytoch
Pytorch中cat和stack的区别
PyTorch中的cat和stack函数都用于连接张量,但方式不同。cat(拼接)不会增加张量维度,而是沿指定维度直接合并数据。例如,两个形状为(2,3)的张量经cat(dim=0)后变为(4,3)。stack(堆叠)则会新增一个维度,将多个张量沿该维度组合。同样两个(2,3)张量经stack(dim=0)后变为(2,2,3)。简言之,cat用于扩展现有维度,stack用于创建新维度。
2023-12-30
62
0
0
30.2℃
深度学习
Pytorch中tensor的维度合并
本文介绍了在PyTorch中使用`view`函数合并tensor维度时保持数据顺序的方法。`view`通过将张量展平后按指定形状重组,但仅当合并连续维度时才能维持原始数据的分类意义(如batch维度)。若需合并非连续维度(例如将`(3,2,4,64)`变为`(3,128,4)`),直接使用`view`会导致数据错乱。此时应先用`permute`调整维度顺序(如变为`(3,2,64,4)`),再使用`view`合并,即可在重组后保持数据顺序与分类逻辑不变。
2023-12-28
115
0
0
35.5℃
深度学习
弹