龙图腾网&IPTOP
- 微信扫码登录
- 账号密码登录
- 短信登录/注册
买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！
申请/专利权人：北京壁仞科技开发有限公司;上海壁仞科技股份有限公司
摘要：本公开提供了一种注意力算子的并行计算方法、处理器、电子设备和介质，包括：获取要进行并行计算的第一张量，第一张量的形状为[B1,S1,H1]；根据处理器的运算单元的数目T、第一轴的数据维度B1、第二轴的数据维度S1，确定将第一张量的第一轴的数据分配为M1份以及将第一张量的第二轴的数据分配为N1份，以使得M1与N1的乘积等于T；按照上述确定，利用处理器的T个运算单元对第一张量进行并行计算。利用根据本公开实施例提供的注意力算子的并行计算方法、处理器、电子设备和介质，能够根据处理器的运算单元的数目、张量的形状综合地确定张量的并行分配策略，以使得充分利用处理器中运算单元的计算资源，提高并行计算性能。
主权项：1.一种用于神经网络中注意力算子的并行计算方法，其特征在于，所述方法适用于包括T个运算单元的处理器，所述方法包括：获取要进行并行计算的第一张量，其中，所述第一张量的形状为[B1,S1,H1]，其中，B1、S1、H1分别表示所述第一张量在第一轴、第二轴和第三轴的数据维度；根据所述处理器的运算单元的数目T、所述第一张量的第一轴的数据维度B1以及所述第一张量的第二轴的数据维度S1，确定将所述第一张量的第一轴的数据分配为M1份以及将所述第一张量的第二轴的数据分配为N1份，以使得M1与N1的乘积等于T；以及按照所述确定，利用所述处理器的所述T个运算单元对所述第一张量进行并行计算，其中，T为大于1的整数，B1、S1、H1为大于等于1的整数。
全文数据：
权利要求：
百度查询： 北京壁仞科技开发有限公司 上海壁仞科技股份有限公司 注意力算子的并行计算方法、处理器、电子设备和介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息，力求客观、公正，但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解，仅供参考使用，不能作为本公司承担任何法律责任的依据或者凭证。
主营中国专利交易买卖与专利转让许可,高校科技成果推广与科技成果转化,知识产权运营管理与平台开发,科技人才专家库与科技猎头等科技服务
开放平台
担保交易
惠及多方
会员特惠
专属平台
适合多方
数据共享
功能齐全
皖公网安备 34010402703815号