[发明专利]最大池化方法、装置、计算机设备及存储介质

权利要求书

1.一种最大池化方法，其特征在于，所述方法包括：

获取特征图，所述特征图由卷积神经网络中的卷积层输出，且所述特征图在内存中采用NCHW内存布局方式，其中，采用所述NCHW内存布局方式时，所述特征图中同一通道内的特征值连续写入内存；

对于所述特征图上的各个池化窗口，获取所述池化窗口内的特征值，其中，所述池化窗口内的特征值从内存中连续读取得到；

根据第一并行计算指令将所述池化窗口内的特征值加载到至少两个寄存器中；

根据第二并行计算指令计算各个所述寄存器中的最大特征值，所述第一并行计算指令和所述第二并行计算指令用于指示进行单指令多数据流SIMD运算；

根据各个所述池化窗口对应的所述最大特征值生成所述特征图的最大池化结果。

2.根据权利要求1所述的方法，其特征在于，所述池化窗口的尺寸的n×n，n为大于等于2的整数；

所述根据第一并行计算指令将所述池化窗口内的特征值加载到至少两个寄存器中，包括：

根据所述第一并行计算指令将所述池化窗口中第i行的特征值加载至第i个寄存器中，i为小于等于n的整数。

3.根据权利要求2所述的方法，其特征在于，所述根据所述第一并行计算指令将所述池化窗口中第i行的特征值加载至第i个寄存器中，包括：

若所述第i个寄存器的尺寸大于等于n，则根据所述第一并行计算指令将所述池化窗口中第i行的特征值加载至所述第i个寄存器中；

其中，当所述第i个寄存器的尺寸大于n时，将所述池化窗口中第i行的特征值加载至所述第i个寄存器后，所述第i个寄存器中的空余位置使用无效值填充。

4.根据权利要求3所述的方法，其特征在于，所述根据第二并行计算指令计算各个所述寄存器中的所述最大特征值，包括：

根据所述第二并行计算指令比较第1个寄存器至第n个寄存器，得到目标寄存器，所述目标寄存器中包含所述第1个寄存器至所述第n个寄存器中对应位置处特征值的最大值；

根据所述目标寄存器确定所述最大特征值。

5.根据权利要求4所述的方法，其特征在于，所述根据所述第二并行计算指令比较第1个寄存器至第n个寄存器，得到目标寄存器，包括：

根据所述第二并行计算指令比较第j个寄存器和第j+1个寄存器中对应位置处的特征值，j大于等于1，且小于等于n-1；

根据比较结果更新所述第j+1个寄存器，其中，更新后的所述第j+1寄存器中包含所述第j个寄存器和所述第j+1个寄存器中对应位置处特征值的最大值；

将更新后的所述第n个寄存器确定为所述目标寄存器。

6.根据权利要求1所述的方法，其特征在于，寄存器为128比特，所述第一并行计算指令为vld1q_f32指令，所述第二并行计算指令为vmaxq_f32指令。

7.一种最大池化装置，其特征在于，所述装置包括：

第一获取模块，用于获取特征图，所述特征图由卷积神经网络中的卷积层输出，且所述特征图在内存中采用NCHW内存布局方式，其中，采用所述NCHW内存布局方式时，所述特征图中同一通道内的特征值连续写入内存；

第二获取模块，用于对于所述特征图上的各个池化窗口，获取所述池化窗口内的特征值，其中，所述池化窗口内的特征值从内存中连续读取得到；

计算模块中的加载单元，用于根据第一并行计算指令将所述池化窗口内的特征值加载到至少两个寄存器中；

所述计算模块中的计算单元，用于根据第二并行计算指令计算各个所述寄存器中的最大特征值，所述第一并行计算指令和所述第一并行计算指令用于指示进行单指令多数据流SIMD运算；

生成模块，用于根据各个所述池化窗口对应的所述最大特征值生成所述特征图的最大池化结果。

8.一种计算机设备，其特征在于，所述计算机设备包括处理器和存储器；所述存储器存储有至少一条指令，所述至少一条指令用于被所述处理器执行以实现如权利要求1至6任一所述的最大池化方法。

9.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有至少一条指令，所述至少一条指令用于被处理器执行以实现如权利要求1至6任一所述的最大池化方法。