当前位置:首页 >> 脚本专栏

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

1、cat:拼接

直接合并数据

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

2、stack拼接:

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

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

3、split拆分:

通过数据长度进行拆分

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

4、chunk拆分:

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

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

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