[发明专利]一种神经网络处理方法、装置、设备及存储介质在审
申请号: | 201910239947.0 | 申请日: | 2019-03-27 |
公开(公告)号: | CN109948790A | 公开(公告)日: | 2019-06-28 |
发明(设计)人: | 孙红岩 | 申请(专利权)人: | 苏州浪潮智能科技有限公司 |
主分类号: | G06N3/063 | 分类号: | G06N3/063 |
代理公司: | 暂无信息 | 代理人: | 暂无信息 |
地址: | 215100 江苏省苏州市吴*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明实施例公开了一种神经网络处理方法、装置、设备及存储介质。其中神经网络处理方法,包括:判断输入cudnn库中的矩阵大小是否为预设值的倍数;如果不是,则将所述矩阵的数据大小进行存储;将所述矩阵的数据作为cudnn的前向卷积函数的输入;将不满足预设值的倍数的空余位置填0补全;进行卷积运算,将卷积运算后的结果,根据存储的数据大小进行重排。本发明实施例在进行神经网络计算时通过扩增矩阵大小,可以图形处理器的输入矩阵条件,使用图形处理器的加速单元进行处理,从而提高了神经网络的处理速度。 | ||
搜索关键词: | 矩阵 神经网络处理 图形处理器 存储介质 卷积运算 预设 存储 神经网络计算 加速单元 空余位置 神经网络 输入矩阵 重排 卷积 扩增 前向 | ||
【主权项】:
1.一种神经网络处理方法,其特征是,包括:判断输入cudnn库中的矩阵大小是否为预设值的倍数;如果不是,则将所述矩阵的数据大小进行存储;将所述矩阵的数据作为cudnn的前向卷积函数的输入;将不满足预设值的倍数的空余位置填0补全;进行卷积运算,将卷积运算后的结果,根据存储的数据大小进行重排。
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于苏州浪潮智能科技有限公司,未经苏州浪潮智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201910239947.0/,转载请声明来源钻瓜专利网。