一种人工智能处理器处理二维复数矩阵的方法和设备.pdf

(19)中华人民共和国国家知识产权局 (12)发明专利申请 (10)申请公布号 CN 111028136 A (43)申请公布日 2020.04.17 (21)申请号 201911349811.1 (22)申请日 2019.12.24 (71)申请人 上海寒武纪信息科技有限公司 地址 201306 上海市浦东新区同汇路168号 B座6层 (72)发明人 不公告发明人　 (74)专利代理机构 北京维昊知识产权代理事务 所(普通合伙) 11804 代理人 李波　孙新国 (51)Int.Cl. G06T 1/60(2006.01) G06T 1/00(2006.01) G06T 1/20(2006.01) 权利要求书4页 说明书20页 附图15页 (54)发明名称 一种人工智能处理器处理二维复数矩阵的 方法和设备 (57)摘要 本公开记载了一种人工智能处理器处理二 维复数矩阵的方法、电子设备和计算装置，其中 计算装置可以包括在组合处理装置中，该组合处 理装置还可以包括通用互联接口和其他处理装 置。所述计算装置与其他处理装置进行交互，共 同完成用户指定的计算操作。组合处理装置还可 以包括存储装置，该存储装置分别与计算装置和 其他处理装置连接，用于计算装置和其他处理装 置的数据。 A 6 3 1 8 2 0 1 1 1 N C CN 111028136 A 权　利　要　求　书 1/4页 1.一种人工智能处理器处理二维复数矩阵的方法，其中，所述二维复数矩阵的尺寸为N ×M；与所述二维复数矩阵对应的行系数矩阵的尺寸为M×M，与所述二维复数矩阵对应的列 系数矩阵的尺寸为N×N，所述方法包括： 根据人工智能处理器上的片上存储单元的第一存储区的容量以及所述二维复数矩阵 的尺寸，将所述二维复数矩阵拆分为至少两个子复数矩阵，所述子复数矩阵的尺寸为n×m， 且所述子复数矩阵的所有元素均能够存放在所述第一存储区内； 将所述子复数矩阵加载至所述人工智能处理器的片上存储单元的第一存储区中； 根据所述子复数矩阵的尺寸，将所述行系数矩阵拆分为至少两个子行系数矩阵，以及 将所述列系数矩阵拆分为至少两个子列系数矩阵，其中，所述子行系数矩阵的尺寸为m×m， 所述子列系数矩阵的尺寸为n×n； 将所述子行系数矩阵加载至所述人工智能处理器的片上存储单元的第二存储区中； 将所述子列系数矩阵加载至所述人工智能处理器的片上存储单元的第三存储区中； 所述人工智能处理器利用所述片上存储单元中的子复数矩阵、子行系数矩阵和子列系 数矩阵来进行傅里叶变换，以得到运算结果；以及 所述人工智能处理器将所述运算结果传送至所述片外存储单元以进行存储。 2.一种人工智能处理器处理二维复数矩阵的方法，其中，所述二维复数矩阵的尺寸为N ×M；与所述二维复数矩阵对应的行系数矩阵的尺寸为M×M，与所述二维复数矩阵对应的列 系数矩阵的尺寸为N×N，所述方法包括： 根据人工智能处理器上的片上存储单元的第二存储区的容量，将所述行系数矩阵拆分 为至少两个子行系数矩阵，所述子行系数矩阵的尺寸为m ×m，并且所述子行系数矩阵的所 有元素均能够存放在所述第二存储区内； 将所述子行系数矩阵加载至所述人工智能处理器上的片上存储单元的第二存储区； 根据所述子行系数矩阵的尺寸，将所述二维复数矩阵拆分为至少两个与所述子行系数 矩阵对应的子复数矩阵，其中所述子复数矩阵的尺寸为n×m； 将所述子复数矩阵加载至所述人工智能处理器上的片上存储单元的第一存储区； 根据所述子复数矩