一種基于阻變存儲器的圖神經(jīng)網(wǎng)絡(luò)計算方法和裝置

基本信息

申請?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 分類 計算；推算；計數(shù);
發(fā)明人 何銀濤;王穎;李華偉;李曉維 申請（專利權(quán)）人 中國科學(xué)院計算技術(shù)研究所
代理機(jī)構(gòu) 北京泛華偉業(yè)知識產(chǎn)權(quán)代理有限公司 代理人 王勇
地址 100190北京市海淀區(qū)中關(guān)村科學(xué)院南路6號
法律狀態(tài) -

摘要

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