合并

cat

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

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

stack

创建一个新的维度,要stack的两个tensor维度要相同!
pytorch中数据合并与拆分--cat,stack,split,chunk

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

拆分

split 根据长度拆分

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

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

chunck 根据数量拆分

pytorch中数据合并与拆分--cat,stack,split,chunk
chunk(2,dim=0)在第0维度上,将数据拆分为2块!

相关文章:

  • 2022-12-23
  • 2021-11-29
  • 2021-10-21
  • 2022-02-17
  • 2022-12-23
  • 2022-01-04
  • 2022-12-23
  • 2021-11-19
猜你喜欢
  • 2022-12-23
  • 2021-10-22
  • 2022-12-23
  • 2021-12-18
  • 2021-07-30
  • 2022-12-23
相关资源
相似解决方案