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

基本信息

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

摘要

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