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

权利要求书

1.一种多输入多输出矩阵最大值池化向量化实现方法，其特征在于，该方法包括：根据向量处理器中向量处理单元VPE的数量以及输入特征图的通道方向将输入的多个输入特征图进行重排序，使得各向量处理单元VPE可以同时进行最大值池化计算，每个向量处理单元VPE独立的计算单个的输出特征图，各向量处理单元VPE每次计算时，向量加载输入特征图的一行数据，依次与其余各行数据进行比较后，同时得到各个输入特征图对应位置池化尺寸的最大值池化结果。

2.根据权利要求1所述的多输入多输出矩阵最大值池化向量化实现方法，其特征在于，该方法的具体步骤包括：

S1.确定向量处理器单核可以同时计算的输入特征图数量；

S2.每次取N张输入特征图中M个输入特征图进行排序，直至完成所有N张输入特征图的排序，得到重排序结果，其中M为向量处理器中向量处理单元VPE的数量；

S3.将所述步骤S2中重排序结果传输至向量处理器核内AM中；

S4.每次向量加载AM中的一行数据，依次与其他行进行最大值比较，得出M个输入特征图的最大值池化结果，最终完成N张输入特征图的最大值池化操作。

3.根据权利要求2所述的多输入多输出矩阵最大值池化向量化实现方法，其特征在于：所述步骤S1中具体根据向量处理器中向量处理单元VPE的数量M、输入特征图的数量N确定向量处理器中单核可以同时计算的输入特征图数量。

4.根据权利要求2所述的多输入多输出矩阵最大值池化向量化实现方法，其特征在于，所述步骤S2中具体将M个输入特征图按照第三维进行排序，执行N/M次后完成所有N张输入特征图的排序，所述第三维为特征图的通道方向；所述按照第三维进行排序具体为在输入特征图的通道方向上，将所有输入特征图同一位置上的元素按顺序放置在一起。

5.根据权利要求2或3或4所述的多输入多输出矩阵最大值池化向量化实现方法，其特征在于，所述步骤S3的具体步骤为：

S31.取AM中输入特征图的一个k×k池化窗口；

S32.在当前池化窗口内向量加载AM中输入特征图的第1行，与第2行进行最大值比较，将比较结果再与第3行进行比较，依此比较k×k次后，可以同时得出M个输入特征图对应位置k×k池化尺寸的取最大值池化结果；

S33.顺移至下一个池化窗口，返回执行步骤S32，直至得到M个输入特征图第2个k×k池化尺寸的最大值池化结果，最终完成N张输入特征图的最大值池化操作。

6.根据权利要求5所述的多输入多输出矩阵最大值池化向量化实现方法，其特征在于：所述步骤S32中具体使用取最大值指令进行最大值比较。

7.根据权利要求5所述的多输入多输出矩阵最大值池化向量化实现方法，其特征在于，所述步骤S33中具体按照移动步长s顺移卷积核窗口，顺移时按照先水平后垂直，以及从左至右、从上到上的顺序移动。

8.根据权利要求1～4中任意一项所述的多输入多输出矩阵最大值池化向量化实现方法，其特征在于，若输入特征图的数量不为VPE数量的整数倍，则将多余的输入特征图使用多个VPE进行处理。