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

合并 cat 要进行cat合并的维度可以不同,但是其余维度必须要相同! stack 创建一个新的维度,要stack的两个tensor维度要相同! 用处:将两个班级的学生和成绩合并,生成维度含义:2个表格,每个表格32个学生,每个学生考了8门试! 拆分 split 根据长度拆分 split([2,1],dim=0):在第0维度上,拆分为两个部分,第一部分是2个,第二部分数量是1个。 chunck 根
相关文章
相关标签/搜索