[发明专利]用于神经网络的池化装置和池化方法

权利要求书

1.一种用于神经网络的池化装置，包括：

池化运算单元，用于对输入的神经元进行池化运算，所述池化运算单元包括求最大值单元、求最小值单元和求平均值单元，所述池化运算单元根据控制单元指示的池化模式选择所述求最大值单元或所述求最小值单元或所述求平均值单元；

控制单元，用于基于所述池化运算单元的处理能力将池化范围内的神经元划分为多个批次并控制各批次的神经元依次输入至所述池化运算单元进行池化运算，所述控制单元通过批次划分编码、池化模式信号以及迭代次数编码来控制池化运算单元的运算过程，所述批次划分编码用于指示将一个池化范围内的神经元划分为多少批次以及每个批次包含的神经元个数，所述池化模式信号用于控制所述池化运算单元所选择的运算模块，所述迭代次数编码用于指示对一个池化范围的神经元需要进行迭代的次数，也即划分的批次数；

内部缓存单元，用于存储所述池化范围内的神经元并根据所述控制单元指示的所述批次划分编码将各批次的神经元依次输入至所述池化运算单元；以及

数据迭代传输单元，用于根据所述控制单元指示的所述迭代次数编码决定将所述池化运算单元的运算结果回传至所述池化运算单元参与下一批次的池化操作或者进行输出。

2.根据权利要求1所述的池化装置，其特征在于，对于所划分的所述池化范围内的神经元的多个批次，其中：

将第一批次的神经元数确定为等于所述池化运算单元的输入接口数；

将第一批次的后续批次的神经元数确定为等于所述池化运算单元的输入接口数减1，直到剩余神经元数小于所述池化运算单元的输入接口数减1；

将所述剩余神经元确定为最后一个批次的神经元。

3.一种基于权利要求1至2中任一项所述池化装置的池化方法，包括以下步骤：

步骤1，用于基于所述池化运算单元的处理能力将池化范围内的神经元划分为多个批次；

步骤2，并对各批次的神经元依次进行池化运算。

4.根据权利要求3所述的池化方法，其中，步骤1包括以下子步骤：

步骤11，第一批次的神经元数确定为等于所述池化运算单元的输入接口数；

步骤12，将第一批次的后续批次的神经元数确定为等于所述池化运算单元的输入接口数减1，直到剩余神经元数小于所述池化运算单元的输入接口数减1；

步骤13，将所述剩余神经元确定为最后一个批次的神经元。

5.根据权利要求3或4所述的池化方法，其中，步骤2包括以下子步骤：

步骤21，对于所划分的所述池化范围内的神经元的多个批次，将第一批次的神经元加载至所述池化运算单元进行池化运算；

步骤22，依次将第一批次的后续各批次的神经元加载至所述池化运算单元并将当前处理批次的前一批次神经元的运算结果回传至所述池化运算单元，直到完成所述池化范围内的所有神经元的池化处理。

6.一种计算机可读存储介质，其上存储有计算机程序，其中，该程序被处理器执行时实现根据权利要求3至5中任一项所述方法的步骤。

7.一种计算机设备，包括存储器和处理器，在所述存储器上存储有能够在处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现权利要求3至5中任一项所述的方法的步骤。