亲宝软件园·资讯

展开

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

杰鑫哥 人气:4

1、cat:拼接

直接合并数据

2、stack拼接:

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

3、split拆分:

通过数据长度进行拆分

4、chunk拆分:

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

加载全部内容

相关教程
猜你喜欢
用户评论