多样本神经网络前向传播向量化实现方法

摘要

本发明公开一种多样本神经网络前向传播向量化实现方法，步骤包括：步骤1：将输入特征数据按照样本维优先的方式存储，以及将权重数据按照输入特征维优先的方式存储；步骤2：按列将输入特征数据集数据矩阵划分为多个矩阵块；步骤3：将权重数据矩阵传输到各个核的SM中，并将每个输入特征数据矩阵传输到各个核的AM中，通过向量化矩阵乘法和并行化矩阵乘法计算；步骤4：各个核将输出特征矩阵与偏置数据列向量进行并行加法计算；步骤5：将输出特征矩阵计算结果存储在向量处理器的片外存储器中；步骤6：重复步骤3至5，直到完成计算。本发明具有实现方法简单、能够充分发挥向量处理器计算性能且实现效率高、带宽需求小以及通用性强等优点。