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

一种张量折叠方法、设备、存储介质及程序产品 

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

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

摘要:本申请实施例提供了一种张量折叠方法、设备、存储介质及程序产品,涉及人工智能技术领域,该方法包括:以折叠后的子张量包含的填充区域最小为基准,从多个候选折叠方式中选取与待处理张量匹配的目标折叠方式。对待处理张量进行填充获得目标填充张量,并按照目标折叠方式将目标填充张量折叠成多个批次的目标子张量进行计算,故针对超出维度大小限制的大型张量,也能进行内存搬运操作,实现对文生图类模型进行性能调优,从而提高模型性能调优的效果。其次,以折叠后的子张量包含的填充区域最小为基准选取目标折叠方式,故对折叠后的多个批次的目标子张量进行计算时,大大减少为填充区域额外分配的计算资源,从而降低人工智能芯片的资源消耗。

主权项:1.一种张量折叠方法,其特征在于,包括:获取多个候选折叠方式,每个候选折叠方式包括:折叠后的子张量的大小;以折叠后的子张量包含的填充区域最小为基准,从所述多个候选折叠方式中,选取与待处理张量匹配的目标折叠方式;对所述待处理张量进行填充,获得目标填充张量,并按照所述目标折叠方式,将所述目标填充张量折叠成多个批次的目标子张量。

全文数据:

权利要求:

百度查询: 上海壁仞科技股份有限公司 一种张量折叠方法、设备、存储介质及程序产品

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