一种用于执行卷积神经网络训练的装置和方法

专利主权项内容

1.一种卷积神经网络训练的装置，包括存储单元、寄存器单元和卷积神经网络反向训练运算单元，其中：存储单元，用于存储输入神经元数据、输出神经元数据、输入神经元数据梯度、输出神经元数据梯度和卷积核；所述寄存器单元，用于存储输入神经元数据存储的地址、输出神经元数据存储的地址、输入神经元数据梯度存储的地址、输出神经元数据梯度存储的地址和卷积核存储的地址；所述卷积神经网络反向训练运算单元，用于依据卷积核，从当前层的正向时输入神经元数据中选择出对应待运算的输入神经元数据，依据卷积窗口，从当前层的前一层的输出数据中选取出对应本层运算模块的输入数据，然后依据这部分输入数据和来自当前层的后一层的数据梯度执行向量乘向量运算，在卷积窗口遍历完数据后，对向量乘向量运算的结果进行求和，得到该卷积核对应的梯度，依据卷积核对应的梯度更新卷积核；所述卷积神经网络反向运算的指令包括：卷积神经网络group指令，根据该指令，分别从数据访问单元的指定地址取出指定大小的输出神经元数据梯度和卷积核，划分group之后，在卷积神经网络反向训练运算部件中做卷积操作，然后将输出结果做激活。