买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:北京壁仞科技开发有限公司;上海壁仞科技股份有限公司
摘要:本公开的实施例公开了一种用于处理张量数据的方法、用于处理张量数据的装置、电子设备与计算机可读存储介质。该方法包括:响应于张量数据在第一轴上的数据长度大于预设阈值,确定第一轴对应的调整倍数;调整张量数据的形状,其中,调整形状后的张量数据在第一轴上的数据长度缩短调整倍数并且在第二轴上的数据长度扩大调整倍数,其中,调整形状后的张量数据在第一轴上的数据长度小于预设阈值;在内存中排布调整形状后的张量数据;以及基于调整倍数,将内存中排布的所述张量数据,按照数据块的粒度加载至计算单元中进行计算。本公开能够在算子的输入数据在某个轴上的长度超过硬件限制时,人工智能芯片仍能高效地处理算子的计算。
主权项:1.一种用于处理张量数据的方法,包括:响应于张量数据在第一轴上的数据长度大于预设阈值,确定所述第一轴对应的调整倍数;调整所述张量数据的形状,其中,调整形状后的所述张量数据在第一轴上的数据长度缩短所述调整倍数并且在第二轴上的数据长度扩大所述调整倍数,其中,调整形状后的张量数据在所述第一轴上的数据长度小于所述预设阈值;在内存中排布所述调整形状后的张量数据;以及基于所述调整倍数,将所述内存中排布的所述张量数据,按照数据块的粒度加载至计算单元中进行计算。
全文数据:
权利要求:
百度查询: 北京壁仞科技开发有限公司 上海壁仞科技股份有限公司 用于处理张量数据的方法和装置
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。