首页 专利交易 科技果 科技人才 科技服务 国际服务 商标交易 会员权益 IP管家助手 需求市场 关于龙图腾
 /  免费注册
到顶部 到底部
清空 搜索

一种分布式自动微分方法及相关装置 

买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!

申请/专利权人:华为技术有限公司

摘要:本申请实施例提供一种分布式自动微分方法及相关装置,方法包括:获取神经网络模型的正向传播计算图;基于计算节点存储的神经网络模型的模型参数与分布式计算系统中除计算节点外的至少一个计算节点存储的模型参数相同,在计算节点的模型参数的输入端和用于计算该模型参数的算子之间插入目标算子以获得目标计算图;目标算子用于透传模型参数,且其导数算法用于对具有相同模型参数的所有计算节点的模型参数的导数进行求和;基于目标计算图中算子的导数算法生成反向传播计算图。实施本申请实施例能避免用户介入模型反向传播过程的推导,提升开发效率,降低开发难度。

主权项:1.一种分布式自动微分方法,所述方法应用于分布式计算系统中的计算节点,其特征在于,包括:获取神经网络模型的正向传播计算图;其中,所述正向传播计算图包括多个算子,所述多个算子中包括本地算子和通信算子,所述本地算子表示可以在所述计算节点中完成计算的算子,所述通信算子表示依赖所述计算节点与所述分布式计算系统中除所述计算节点外的至少一个计算节点之间的通信来完成计算的算子;基于所述计算节点存储的神经网络模型的模型参数与所述分布式计算系统中除所述计算节点外的至少一个计算节点存储的模型参数相同,在所述计算节点的模型参数的输入端和用于计算该模型参数的算子之间插入目标算子,以获得目标计算图;其中,所述目标算子用于透传模型参数,且所述目标算子的导数算法用于对具有相同模型参数的所有计算节点的模型参数的导数进行求和;基于所述目标计算图中部分或全部算子的导数算法来生成所述神经网络模型的反向传播计算图。

全文数据:

权利要求:

百度查询: 华为技术有限公司 一种分布式自动微分方法及相关装置

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。