基于廣播機(jī)制進(jìn)行張量計(jì)算的方法、裝置、芯片及介質(zhì)
基本信息
| 申請(qǐng)?zhí)?/td> | CN202011528418.1 | 申請(qǐng)日 | - |
| 公開（公告）號(hào) | CN112597424A | 公開（公告）日 | 2021-04-02 |
| 申請(qǐng)公布號(hào) | CN112597424A | 申請(qǐng)公布日 | 2021-04-02 |
| 分類號(hào) | G06F17/10(2006.01)I | 分類 | 計(jì)算；推算；計(jì)數(shù); |
| 發(fā)明人 | 吳欣洋;李涵;戚海濤;丁瑞強(qiáng);馮開革 | 申請(qǐng)（專利權(quán)）人 | 無錫靈汐類腦科技有限公司 |
| 代理機(jī)構(gòu) | 北京品源專利代理有限公司 | 代理人 | 孟金喆 |
| 地址 | 214028 江蘇省無錫市新吳區(qū)弘毅路10號(hào)金乾座1701-1710室 | ||
| 法律狀態(tài) | - | ||
摘要
| 摘要 | 本發(fā)明實(shí)施例公開了一種基于廣播機(jī)制進(jìn)行張量計(jì)算的方法、裝置、芯片及介質(zhì)。該方法包括：獲取第一張量和第二張量，第一張量的維數(shù)大于或等于第二張量的維數(shù)；判斷第一張量和第二張量是否滿足廣播機(jī)制條件，若滿足，則確定與第一張量和第二張量分別對(duì)應(yīng)的第一目標(biāo)張量和第二目標(biāo)張量，第一目標(biāo)張量和第二目標(biāo)張量的后緣維度的軸長度相符；通過通用計(jì)算硬件單元組，采用廣播機(jī)制對(duì)第一目標(biāo)張量和第二目標(biāo)張量進(jìn)行張量計(jì)算，得到張量計(jì)算備選結(jié)果；根據(jù)張量計(jì)算備選結(jié)果，確定與第一張量和第二張量對(duì)應(yīng)的目標(biāo)計(jì)算結(jié)果。上述技術(shù)方案實(shí)現(xiàn)了在無支持廣播計(jì)算的專用底層硬件時(shí)全面支持廣播機(jī)制中所有形式的廣播操作。?? |
