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

权利要求书

1.一种人工智能处理器处理二维数据矩阵的方法，包括：

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

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

所述人工智能处理器对所述子矩阵进行矩阵转置运算，以得到运算结果；以及

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

2.根据权利要求1所述的方法，其中，将所述二维数据矩阵拆分为至少两个子矩阵包括：

根据所述二维数据矩阵的尺寸的不同，来动态地调整所述子矩阵的尺寸。

3.根据权利要求2所述的方法，其中，动态地调整所述子矩阵的尺寸包括：

设定所述子矩阵的初始行数和初始列数；

确定所述二维数据矩阵的行数与所述初始行数之间的第一比值；

确定所述二维数据矩阵的列数与所述初始列数之间的第二比值；

调整所述子矩阵的行数和列数，以使得所述第一比值和第二比值均为整数。

4.根据权利要求3所述的方法，其中，当所述第一比值和/或第二比值为非整数时，对所述二维数据矩阵的行/或列进行扩展，以使得所述第一比值和/或所述第二比值为整数。

5.根据权利要求2-4中任意一项所述的方法，其中，所述子矩阵之间彼此紧邻地设置。

6.根据权利要求1所述的方法，其中，所述子矩阵的尺寸是固定的，其具有预定行数和预定列数。

7.根据权利要求6所述的方法，其中，

确定所述二维数据矩阵的行数与所述预定行数之间的第三比值；

确定所述二维数据矩阵的列数与所述预定列数之间的第四比值；

当所述第三比值和/或第四比值为非整数时，对所述二维数据矩阵的行和/或列进行扩展，以使得所述所述第三比值和/或所述第四比值为整数。

8.根据权利要求6所述的方法，其中，所述子矩阵中的至少两个矩阵存在重叠部分。

9.根据权利要求8所述的方法，进一步包括：标识所述重叠部分，以对所述重叠部分进行定位。

10.根据权利要求1-9中任意一项所述的方法，进一步包括：在所述片外存储单元中，根据所述子矩阵在所述二维数据矩阵中的坐标，逐行或逐列地存储所述子矩阵，以形成一维数组。

11.根据权利要求10所述的方法，其中，根据子矩阵的编号、子矩阵的尺寸以及元素在子矩阵中的绝对位置，来计算所述元素在所述一维数组中的位置。

12.根据权利要求11所述的方法，其中，计算所述元素在所述一维数组中的位置包括：

以所述子矩阵的编号乘以所述子矩阵的尺寸，再加上所述元素在所述子矩阵中的绝对位置。

13.根据权利要求1-12中任意一项所述的方法，其中，所述人工智能处理器将所述运算结果传送至所述片外存储单元以进行存储包括：

将经过转置的多个子矩阵存储到片外存储单元中，从而所述多个转置子矩阵构成转置二维数据矩阵。

14.根据权利要求13所述的方法，其中，如果多个子矩阵之间存在重叠部分，在对所述重叠部分进行均值处理。

15.根据权利要求1-14中任意一项所述的方法，其中，所述人工智能处理器对所述子矩阵进行矩阵转置运算，以得到运算结果包括：

通过多个人工智能处理器分别对多个子矩阵进行并行矩阵转置运算。