[发明专利]多输入多输出矩阵平均值池化向量化实现方法

权利要求书

1.一种多输入多输出矩阵平均值池化的向量化实现方法，其特征在于，步骤为：

S1：根据向量处理器中向量处理单元VPE的数量M，输入特征图的数量N，尺寸为n×n，平均值池化的移动步长为s，池化窗口k，确定向量处理器单核同时计算的输入特征图数量；

S2：将M个输入特征图按照第三维进行排序；

S3：步骤S2进行N/M次，直至完成所有N张输入特征图的排序；

S4：将步骤S3中排序好的输入特征图由DMA传输至向量处理器核内AM中；

S5：向量加载AM中的第1行，依次与接下来的数据行进行累加，共计累加k×k次，并将累加结果与1/k²相乘，同时得出M个输入特征图对应位置池化窗口为k×k的取平均值池化结果；

S6：根据水平移动步长s，顺移至下一个池化窗口，进行同步骤S5的计算，得出M个输入特征图第2个k×k池化窗口的平均值池化结果；

S7：重复步骤S5-S6直至完成整个M个输入特征图所有池化窗口的平均值池化操作；

S8：重复步骤S6-S7N/M次，最终完成N张输入特征图的平均值池化操作。

2.根据权利要求1所述的多输入多输出矩阵平均值池化的向量化实现方法，其特征在于，所述步骤S1中，输入特征图的数量N远大于向量处理单元VPE的数量M，且N为M的整数倍。

3.根据权利要求1所述的多输入多输出矩阵平均值池化的向量化实现方法，其特征在于，所述步骤S1中，输入特征图是一个三维矩阵，即高度、宽度、数量。

4.根据权利要求3所述的多输入多输出矩阵平均值池化的向量化实现方法，其特征在于，所述步骤S1中的三维矩阵为方阵，即高度和宽度相等。

5.根据权利要求1-4中任意一项所述的多输入多输出矩阵平均值池化的向量化实现方法，其特征在于，所述步骤S1中平均值池化的移动步长分为水平移动步长和垂直移动步长。

6.根据权利要求5所述的多输入多输出矩阵平均值池化的向量化实现方法，其特征在于，所述步骤S1中取水平移动步长和垂直移动步长相同。

7.根据权利要求1-4中任意一项所述的多输入多输出矩阵平均值池化的向量化实现方法，其特征在于，所述步骤S1中的池化尺寸取方阵，即最大值池化窗口为k×k。

8.根据权利要求1-4中任意一项所述的多输入多输出矩阵平均值池化的向量化实现方法，其特征在于，所述步骤S1中，若输入特征图的数量不为VPE数量的整数倍，则多余的输入特征图由若干VPE进行处理。

9.根据权利要求1-4中任意一项所述的多输入多输出矩阵平均值池化的向量化实现方法，其特征在于，所述步骤S6中，输入特征图每次取一个与k×k对应大小的池化窗口，且该窗口根据移动步长先水平移动，后垂直移动，移动顺序为，从左到右，从上到下。