一種基于阻變存儲器的圖神經網絡計算方法和裝置

基本信息

申請?zhí)?/td> CN202110856642.1 申請日 -
公開（公告）號 CN114186598A 公開（公告）日 2022-03-15
申請公布號 CN114186598A 申請公布日 2022-03-15
分類號 G06K9/62（2022.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數;
發(fā)明人 何銀濤;王穎;李華偉;李曉維 申請（專利權）人 中國科學院計算技術研究所
代理機構 北京泛華偉業(yè)知識產權代理有限公司 代理人 王勇
地址 100190北京市海淀區(qū)中關村科學院南路6號
法律狀態(tài) -

摘要

摘要 本發(fā)明實施例提供了一種基于阻變存儲器的圖神經網絡計算方法和裝置，該方法包括：對于圖神經網絡的任一層，分析該層中將要在阻變存儲器陣列中運算的圖數據在權重原位計算模式和混合原位計算模式下的處理時延相對大小，選擇時延最小的模式作為該層的計算模式；在權重原位計算模式，對所述圖神經網絡的所述層將圖數據的鄰接矩陣和圖神經網絡的權重參數作為原位數據分別映射到相應的阻變存儲器陣列中，以將圖神經網絡的節(jié)點特征作為輸入數據與相應的原位數據進行運算；在混合原位計算模式，對所述圖神經網絡的所述層將圖數據的鄰接矩陣和節(jié)點特征作為原位數據分別映射到相應的阻變存儲器陣列中，以將權重參數作為輸入數據與相應的原位數據進行運算。