pytorch数据拼接与拆分

1、cat:拼接

直接合并数据

pytorch数据拼接与拆分

2、stack拼接:

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

pytorch数据拼接与拆分

 3、split拆分:

通过数据长度进行拆分

pytorch数据拼接与拆分

4、chunk拆分:

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

pytorch数据拼接与拆分

相关推荐