基于廣播機制進行張量計算的方法、裝置、芯片及介質(zhì)

基本信息

申請?zhí)?/td> CN202011528418.1 申請日 -
公開（公告）號 CN112597424A 公開（公告）日 2021-04-02
申請公布號 CN112597424A 申請公布日 2021-04-02
分類號 G06F17/10(2006.01)I 分類 計算；推算；計數(shù);
發(fā)明人 吳欣洋;李涵;戚海濤;丁瑞強;馮開革 申請（專利權(quán)）人 無錫靈汐類腦科技有限公司
代理機構(gòu) 北京品源專利代理有限公司 代理人 孟金喆
地址 214028 江蘇省無錫市新吳區(qū)弘毅路10號金乾座1701-1710室
法律狀態(tài) -

摘要

摘要 本發(fā)明實施例公開了一種基于廣播機制進行張量計算的方法、裝置、芯片及介質(zhì)。該方法包括：獲取第一張量和第二張量，第一張量的維數(shù)大于或等于第二張量的維數(shù)；判斷第一張量和第二張量是否滿足廣播機制條件，若滿足，則確定與第一張量和第二張量分別對應(yīng)的第一目標張量和第二目標張量，第一目標張量和第二目標張量的后緣維度的軸長度相符；通過通用計算硬件單元組，采用廣播機制對第一目標張量和第二目標張量進行張量計算，得到張量計算備選結(jié)果；根據(jù)張量計算備選結(jié)果，確定與第一張量和第二張量對應(yīng)的目標計算結(jié)果。上述技術(shù)方案實現(xiàn)了在無支持廣播計算的專用底層硬件時全面支持廣播機制中所有形式的廣播操作。??