[发明专利]一种人工智能处理器处理二维复数矩阵的方法和设备

权利要求书

1.一种人工智能处理器处理二维复数矩阵的方法，其中，所述二维复数矩阵的尺寸为N×M；与所述二维复数矩阵对应的行系数矩阵的尺寸为M×M，与所述二维复数矩阵对应的列系数矩阵的尺寸为N×N，所述方法包括：

根据人工智能处理器上的片上存储单元的第一存储区的容量以及所述二维复数矩阵的尺寸，将所述二维复数矩阵拆分为至少两个子复数矩阵，所述子复数矩阵的尺寸为n×m，且所述子复数矩阵的所有元素均能够存放在所述第一存储区内；

将所述子复数矩阵加载至所述人工智能处理器的片上存储单元的第一存储区中；

根据所述子复数矩阵的尺寸，将所述行系数矩阵拆分为至少两个子行系数矩阵，以及将所述列系数矩阵拆分为至少两个子列系数矩阵，其中，所述子行系数矩阵的尺寸为m×m，所述子列系数矩阵的尺寸为n×n；

将所述子行系数矩阵加载至所述人工智能处理器的片上存储单元的第二存储区中；

将所述子列系数矩阵加载至所述人工智能处理器的片上存储单元的第三存储区中；

所述人工智能处理器利用所述片上存储单元中的子复数矩阵、子行系数矩阵和子列系数矩阵来进行傅里叶变换，以得到运算结果；以及

所述人工智能处理器将所述运算结果传送至所述片外存储单元以进行存储。

2.一种人工智能处理器处理二维复数矩阵的方法，其中，所述二维复数矩阵的尺寸为N×M；与所述二维复数矩阵对应的行系数矩阵的尺寸为M×M，与所述二维复数矩阵对应的列系数矩阵的尺寸为N×N，所述方法包括：

根据人工智能处理器上的片上存储单元的第二存储区的容量，将所述行系数矩阵拆分为至少两个子行系数矩阵，所述子行系数矩阵的尺寸为m×m，并且所述子行系数矩阵的所有元素均能够存放在所述第二存储区内；

将所述子行系数矩阵加载至所述人工智能处理器上的片上存储单元的第二存储区；

根据所述子行系数矩阵的尺寸，将所述二维复数矩阵拆分为至少两个与所述子行系数矩阵对应的子复数矩阵，其中所述子复数矩阵的尺寸为n×m；

将所述子复数矩阵加载至所述人工智能处理器上的片上存储单元的第一存储区；

根据所述子复数矩阵，将所述子列系数矩阵拆分为多至少两个与所述子复数矩阵对应的子列系数矩阵，其中所述子列系数矩阵的尺寸为n×n；

将所述子列系数矩阵加载至所述人工智能处理器上的片上存储单元的第三存储区；

所述人工智能处理器利用所述片上存储单元中的子复数矩阵、子行系数矩阵和子列系数矩阵来进行傅里叶变换，以得到运算结果；以及

所述人工智能处理器将所述运算结果传送至所述片外存储单元以进行存储。

3.一种人工智能处理器处理二维复数矩阵的方法，其中，所述二维复数矩阵的尺寸为N×M；与所述二维复数矩阵对应的行系数矩阵的尺寸为M×M，与所述二维复数矩阵对应的列系数矩阵的尺寸为N×N，所述方法包括：

根据人工智能处理器上的片上存储单元的第三存储区的容量，将所述列系数矩阵拆分为至少两个子列系数矩阵，所述子列系数矩阵的尺寸为n×n，并且所述子列系数矩阵的所有元素均能够存放在所述第三存储区内；

将所述子列系数矩阵加载至所述人工智能处理器上的片上存储单元的第三存储区；

根据所述子列系数矩阵，将所述二维复数矩阵拆分为至少两个与所述子列系数矩阵对应的子复数矩阵，其中所述子复数矩阵的尺寸为n×m；

将所述子复数矩阵加载至所述人工智能处理器上的片上存储单元的第一存储区；

根据所述子复数矩阵，将所述子行系数矩阵拆分为至少两个与所述子复数矩阵对应的子行系数矩阵，其中所述子行系数矩阵的尺寸为m×m；

将所述子行系数矩阵加载至所述人工智能处理器上的片上存储单元的第二存储区；

所述人工智能处理器利用所述片上存储单元中的子复数矩阵、子行系数矩阵和子列系数矩阵来进行傅里叶变换，以得到运算结果；以及

所述人工智能处理器将所述运算结果传送至所述片外存储单元以进行存储。

4.根据权利要求1-3中任意一项所述的方法，其中，所述人工智能处理器将所述子行系数矩阵转换为一维数组以加载至所述第二存储区；将所述子列系数矩阵转换为一维数组以加载至所述第三存储区。