-
cat 合并
dim必须首选相同(上例都是3),其次除了合并的dim(上例中为dim0)外,其它dim的size必须相同(dim 1的size是32,dim2的size是8),否则需要手动处理到相同再合并。 -
stack
stack叠加,会增加一个新维度,下图显示stack与cat的差别
-
split 按长度拆分
-
chunk 按数量拆分
pytorch学习笔记6 tensor拼接和拆分
最新推荐文章于 2024-08-07 20:02:50 发布
cat 合并
dim必须首选相同(上例都是3),其次除了合并的dim(上例中为dim0)外,其它dim的size必须相同(dim 1的size是32,dim2的size是8),否则需要手动处理到相同再合并。
stack
stack叠加,会增加一个新维度,下图显示stack与cat的差别
split 按长度拆分
chunk 按数量拆分