一種基于位掩碼的卷積壓縮計(jì)算方法及系統(tǒng)

基本信息

申請(qǐng)?zhí)?/td> CN202110134844.5 申請(qǐng)日 -
公開(kāi)（公告）號(hào) CN112949851A 公開(kāi)（公告）日 2021-06-11
申請(qǐng)公布號(hào) CN112949851A 申請(qǐng)公布日 2021-06-11
分類(lèi)號(hào) G06N3/08;G06F7/544;G06F16/901;G06F17/16 分類(lèi) 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 張笑;林廣棟;黃光紅;顧大曄;劉國(guó)成 申請(qǐng)（專(zhuān)利權(quán)）人 安徽芯紀(jì)元科技有限公司
代理機(jī)構(gòu) 合肥律眾知識(shí)產(chǎn)權(quán)代理有限公司 代理人 鄧盛花
地址 230000 安徽省合肥市高新區(qū)習(xí)友路3366號(hào)博微產(chǎn)業(yè)園系統(tǒng)協(xié)同中心7樓
法律狀態(tài) -

摘要

摘要 本發(fā)明提出一種基于位掩碼的卷積壓縮計(jì)算方法及系統(tǒng)，輸入矩陣F大小為h*w，卷積核K大小為k1h*k2w，卷積壓縮計(jì)算方法包括以下步驟：1、提取輸入矩陣F的非0數(shù)據(jù)形成行矩陣Df，提取輸入矩陣F的位掩碼形成矩陣Mf，提取卷積核K的非0數(shù)據(jù)形成行矩陣Dk，提取卷積核K的位掩碼形成矩陣Mk；2、將行矩陣Dk中的每個(gè)元素與行矩陣Df中的元素逐一相乘，得到多個(gè)乘法結(jié)果矩陣；3、利用矩陣Mk和矩陣Mf將每個(gè)乘法結(jié)果矩陣恢復(fù)成卷積核K大小的恢復(fù)矩陣；4、對(duì)每個(gè)乘法結(jié)果矩陣得到的恢復(fù)矩陣進(jìn)行對(duì)位累加，得到卷積計(jì)算結(jié)果。本發(fā)明可以避免卷積運(yùn)算過(guò)程中0值參與乘法計(jì)算，降低了算力和功耗的浪費(fèi)，提高了卷積計(jì)算速度。