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

用于生成二进制掩码张量的方法、计算设备和存储介质 

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

申请/专利权人:北京壁仞科技开发有限公司;上海壁仞科技股份有限公司

摘要:本发明的实施例涉及一种用于生成二进制掩码张量的方法、计算设备和存储介质。该方法包括:接收指示待生成的二进制掩码张量的概率的第一指令;针对待生成的二进制掩码张量在第一维度中的每一行,确定随机种子;至少基于概率和随机种子,确定待生成的二进制掩码张量在第二维度上的元素的值,以便生成二进制掩码张量。根据本发明提供的用于生成二进制掩码张量的方法,能够并行地确定二进制掩码张量中各行的元素,从而提高二进制掩码张量的生成效率。

主权项:1.一种用于生成适用于AI模型训练的二进制掩码张量的方法,其特征在于,所述AI模型包括神经网络层,所述方法包括:接收指示待生成的二进制掩码张量的概率的第一指令;响应于接收所述第一指令,基于所述第一指令生成初始掩码张量,其中所述初始掩码张量中的所有元素的初始值相同;针对所述待生成的二进制掩码张量在第一维度中的每一行,确定随机种子;至少基于所述概率和所述随机种子,并行地确定所述待生成的二进制掩码张量在第二维度上的元素的值;基于所确定的所述待生成的二进制掩码张量在所述第二维度上的元素的值,确定所述初始掩码张量中需要更新的元素的位置;以及响应于确定所述初始掩码张量中需要更新的元素的位置,对相应的元素的初始值进行更新,以便以批量的方式生成二进制掩码张量,其中所述第一维度为所述二进制掩码张量的高度维度,所述第二维度为所述二进制掩码张量的宽度维度。

全文数据:

权利要求:

百度查询: 北京壁仞科技开发有限公司 上海壁仞科技股份有限公司 用于生成二进制掩码张量的方法、计算设备和存储介质

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

相关技术
相关技术
相关技术