一种大规模矩阵卷积的多核实现方法

摘要

一种大规模矩阵卷积的多核实现方法，其步骤为：S1：根据多核处理器的核数M，单核VPE的数量为P，输入特征图的尺寸W×H×C，卷积核的尺寸k×k，通道数为C，卷积核的数量N，水平或垂直移动步长s，确定输入特征图的多核划分方法；S2：将输入特征图按行进行划分，将卷积核k×k×C×N按N方向进行划分，单核分到的卷积核数量为P；S3：将按行划分后的输入特征图分给若干计算核，将划分后的卷积核分给相应的核；S4：所有核同时进行同样的操作，直至完成整个输入特征图的卷积操作。本发明具有实现简单、操作方便、可提高大规模矩阵卷积并行性、可以充分利用多核处理器加速等优点。