pytorch中数据合并与拆分--cat,stack,split,chunk

合并

cat

在这里插入图片描述

要进行cat合并的维度可以不同,但是其余维度必须要相同!

stack

创建一个新的维度,要stack的两个tensor维度要相同!
在这里插入图片描述

用处:将两个班级的学生和成绩合并,生成维度含义:2个表格,每个表格32个学生,每个学生考了8门试!

拆分

split 根据长度拆分

split([2,1],dim=0):在第0维度上,拆分为两个部分,第一部分是2个,第二部分数量是1个。

在这里插入图片描述

chunck 根据数量拆分

在这里插入图片描述
chunk(2,dim=0)在第0维度上,将数据拆分为2块!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章