Pytorch数据拼接与拆分操作实现图解

1、cat：拼接

直接合并数据

2、stack拼接：

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

3、split拆分：

通过数据长度进行拆分

4、chunk拆分：

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

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。