[发明专利]一种卷积神经网络中计算资源的调度方法及装置

权利要求书

1.一种卷积神经网络中计算资源的调度方法，其特征在于，所述方法包括：

1)、获取卷积神经网络中的原始矩阵以及核矩阵；

2)、根据核矩阵的尺寸从原始矩阵中获取当前子矩阵，逐行分别将当前子矩阵中的每一行以及核矩阵调度至一个目标乘法器进行计算，或者逐列分别将当前子矩阵中的每一列以及核矩阵调度至一个目标乘法器进行计算，直至当前子矩阵中的元素被调度完；再根据当前子矩阵获取方法获取当前子矩阵的在行方向上或者列方向上的下一个子矩阵，并将该子矩阵作为当前子矩阵，并返回执行所述逐行分别将当前子矩阵中的每一行以及核矩阵调度至一个目标乘法器进行计算，或者逐列分别将当前子矩阵中的每一列以及核矩阵调度至一个目标乘法器进行计算的步骤；

3)、根据所述乘法器的运算结果，与卷积后的矩阵中的元素位置的对应关系，将所述乘法器的运算结果组合成卷积后的矩阵。

2.根据权利要求1所述的一种卷积神经网络中计算资源的调度方法，其特征在于，所述当前子矩阵获取方法包括：

A：从所述原始矩阵的起始端开始，根据所述核矩阵的行数和列数，读取原始矩阵中与核矩阵的列数相同列数的元素，将所读取的元素组成的矩阵作为当前子矩阵；

B：判断针对原始矩阵的调度方式是按照行方向进行调度或者列方向进行调；

C：若按照行方向进行调度，将当前子矩阵之后一列中与当前子矩阵位于相同行的元素作为新元素，将所述当前子矩阵中除离新元素距离最远的一列之外的其他元素，和所述新元素构成的新的子矩阵作为当前子矩阵；

或者将当前子矩阵之前一列中与当前子矩阵位于相同行的元素作为新元素，将所述当前子矩阵中除离新元素距离最远的一列之外的其他元素，和所述新元素构成的新的子矩阵作为当前子矩阵；

D：若按照列方向进行调度，将当前子矩阵之后一行中与当前子矩阵位于相同列的元素作为新元素，将所述当前子矩阵中除离新元素距离最远的一行之外的其他元素，和所述新元素构成的新的子矩阵作为当前子矩阵；

或者将当前子矩阵之前一行中与当前子矩阵位于相同列的元素作为新元素，将所述当前子矩阵中除离新元素距离最远的一行之外的其他元素，和所述新元素构成的新的子矩阵作为当前子矩阵。

3.根据权利要求2所述的一种卷积神经网络中计算资源的调度方法，其特征在于，若当前子矩阵中存在新元素，在对当前子矩阵中的元素进行调度时，仅将新元素中的对应行或者列的元素进行调度。

4.一种卷积神经网络中计算资源的调度装置，其特征在于，所述装置包括：

获取模块，用于获取卷积神经网络中的原始矩阵以及核矩阵；

运算模块，用于根据核矩阵的尺寸从原始矩阵中获取当前子矩阵，逐行分别将当前子矩阵中的每一行以及核矩阵调度至一个目标乘法器进行计算，或者逐列分别将当前子矩阵中的每一列以及核矩阵调度至一个目标乘法器进行计算，直至当前子矩阵中的元素被调度完；再根据子矩阵获取方法获取当前子矩阵的在行方向上或者列方向上的下一个子矩阵，并将该子矩阵作为当前子矩阵，并返回执行所述逐行分别将当前子矩阵中的每一行以及核矩阵调度至一个目标乘法器进行计算，或者逐列分别将当前子矩阵中的每一列以及核矩阵调度至一个目标乘法器进行计算的步骤；

组合模块，用于根据所述乘法器的运算结果，与卷积后的矩阵中的元素位置的对应关系，将所述乘法器的运算结果组合成卷积后的矩阵。