一种基于位掩码的卷积压缩计算方法及系统
基本信息
申请号 | CN202110134844.5 | 申请日 | - |
公开(公告)号 | CN112949851A | 公开(公告)日 | 2021-06-11 |
申请公布号 | CN112949851A | 申请公布日 | 2021-06-11 |
分类号 | G06N3/08;G06F7/544;G06F16/901;G06F17/16 | 分类 | 计算;推算;计数; |
发明人 | 张笑;林广栋;黄光红;顾大晔;刘国成 | 申请(专利权)人 | 安徽芯纪元科技有限公司 |
代理机构 | 合肥律众知识产权代理有限公司 | 代理人 | 邓盛花 |
地址 | 230000 安徽省合肥市高新区习友路3366号博微产业园系统协同中心7楼 | ||
法律状态 | - |
摘要
摘要 | 本发明提出一种基于位掩码的卷积压缩计算方法及系统,输入矩阵F大小为h*w,卷积核K大小为k1h*k2w,卷积压缩计算方法包括以下步骤:1、提取输入矩阵F的非0数据形成行矩阵Df,提取输入矩阵F的位掩码形成矩阵Mf,提取卷积核K的非0数据形成行矩阵Dk,提取卷积核K的位掩码形成矩阵Mk;2、将行矩阵Dk中的每个元素与行矩阵Df中的元素逐一相乘,得到多个乘法结果矩阵;3、利用矩阵Mk和矩阵Mf将每个乘法结果矩阵恢复成卷积核K大小的恢复矩阵;4、对每个乘法结果矩阵得到的恢复矩阵进行对位累加,得到卷积计算结果。本发明可以避免卷积运算过程中0值参与乘法计算,降低了算力和功耗的浪费,提高了卷积计算速度。 |
