Pytorch数据拼接与拆分操作实现图解
作者:杰鑫哥
这篇文章主要介绍了Pytorch数据拼接与拆分操作实现图解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
1、cat:拼接
直接合并数据
2、stack拼接:
与cat不同的是,stack创建了一个新的维度,在拼接的同时,给数据增加了类别。并且stack的所有数据维度必须一致。
3、split拆分:
通过数据长度进行拆分
4、chunk拆分:
与split不同的是:chunk是指定拆分的个数,将数据拆分为指定个数。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。