[发明专利]一种计算装置及相关产品在审
申请号: | 201810161636.2 | 申请日: | 2018-02-27 |
公开(公告)号: | CN110196735A | 公开(公告)日: | 2019-09-03 |
发明(设计)人: | 不公告发明人 | 申请(专利权)人: | 上海寒武纪信息科技有限公司 |
主分类号: | G06F9/302 | 分类号: | G06F9/302;G06F9/38;G06F13/28 |
代理公司: | 广州三环专利商标代理有限公司 44202 | 代理人: | 郝传鑫;熊永强 |
地址: | 200120 上海市浦*** | 国省代码: | 上海;31 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本公开提供了一种计算装置,所述计算装置获取训练样本、训练模型;所述训练模型包括n层结构;包括:指令控制单元用于获取训练指令,根据所述训练指令得到正向运算指令和反向运算指令;运算单元用于根据所述正向运算指令对所述训练样本进行n层正向运算得到第n层正向运算结果,根据所述第n层正向运算结果获得第n层输出数据梯度,根据所述反向运算指令对所述第n层输出数据梯度执行n层反向运算得到所述训练模型的权值梯度;所述训练模型的权值梯度包括每层的权值梯度;压缩单元用于对所述训练模型的权值梯度进行处理,以对应得到处理后的权值梯度;计算装置用于依据所述处理后的权值梯度对所述训练模型的权值进行更新,以完成训练。 | ||
搜索关键词: | 训练模型 正向 计算装置 运算指令 输出数据 训练样本 运算结果 运算 指令 指令控制单元 压缩单元 运算单元 更新 | ||
【主权项】:
1.一种计算装置,其特征在于,包括运算单元、指令控制单元以及压缩单元;所述计算装置获取训练样本、训练模型;所述训练模型为:神经网络训练模型和/或非神经网络训练模型,所述训练模型包括n层结构;所述指令控制单元用于获取训练指令,根据所述训练指令得到正向运算指令和反向运算指令;将所述正向运算指令和所述反向运算指令分别发送给所述运算单元;所述运算单元用于根据所述正向运算指令对所述训练样本进行n层正向运算得到第n层正向运算结果;根据所述第n层正向运算结果获得第n层输出数据梯度;所述运算单元还用于根据所述反向运算指令对所述第n层输出数据梯度执行n层反向运算得到所述训练模型的权值梯度;所述训练模型的权值梯度包括n层中每层的权值梯度;所述压缩单元用于对所述训练模型的权值梯度进行处理,以对应得到处理后的权值梯度;所述计算装置用于依据所述处理后的权值梯度对所述训练模型的权值进行更新,以完成训练。
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于上海寒武纪信息科技有限公司,未经上海寒武纪信息科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201810161636.2/,转载请声明来源钻瓜专利网。
- 上一篇:一种计算装置及相关产品
- 下一篇:电子设备及其操作方法