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

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