[发明专利]一种面向稀疏二维卷积的运算方法和系统

摘要

本发明涉及一种面向稀疏二维卷积的运算方法和系统，利用稀疏二维卷积具有分块计算的特性，将稀疏二维卷积转化为规则的二维卷积，使得不用对原始的运算单元进行侵入式地修改，只需要增加一个控制单元，就能消除稀疏二维卷积的冗余计算，从而获得更高的计算效率。

主权项

1.一种面向稀疏二维卷积的运算方法，其特征在于，包括：步骤1，获取运算单元中乘累加阵列的阵列行数和列数，以将输入特征图存储至输入特征缓存，将卷积核经过压缩处理后存储至卷积核缓存，并将初始化的输出特征图存储至输出特征图缓存，Pk指针指向该卷积核缓存中的第一行，Pi指针指向该输入特征图缓存中的第一行；步骤2，根据卷积配置参数、该Pk指针指向的索引值、该Pk指针指向的地址和该Pi指针指向的地址，得到输出特征图缓存的地址，该卷积配置参数包括：该阵列行数、该阵列列数、该输入特征图的尺寸和该卷积核的尺寸；步骤3，根据输出特征图缓存的地址，读取该输出特征图缓存中的数据，并将其存储至输出矩阵寄存器组；步骤4，该Pk指针指向的数据值和该Pi指针指向的数据值进行标量矩阵乘法，所得结果再与该输出矩阵寄存器组中数据进行矩阵加法，所得结果根据输出特征图数据的地址写回至该输出特征图缓存；步骤5，将Pk指向下一行，判断该Pk指针指向的缓存行是否存有卷积核数据，若是，则执行步骤2，否则将该Pk指针指向卷积核缓存中的第一行；步骤6，将Pi指向下一行，判断该Pi指针指向的缓存行是否存储有输入特征图数据，若是，则执行步骤2，否则输出该输出特征图缓存，计算结束。