pytorch数据拼接与拆分

Posted 2021-03-03 jaysonteng

1、cat：拼接

直接合并数据

2、stack拼接：

与cat不同的是，stack创建了一个新的维度，在拼接的同时，给数据增加了类别。并且stack的所有数据维度必须一致。

 

 3、split拆分：

通过数据长度进行拆分

 

4、chunk拆分：

与split不同的是：chunk是指定拆分的个数，将数据拆分为指定个数。