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