張量切分方法、裝置、芯片及介質(zhì)
基本信息
申請(qǐng)?zhí)?/td> | CN202011479734.4 | 申請(qǐng)日 | - |
公開(公告)號(hào) | CN112507173A | 公開(公告)日 | 2021-03-16 |
申請(qǐng)公布號(hào) | CN112507173A | 申請(qǐng)公布日 | 2021-03-16 |
分類號(hào) | G06F16/901(2019.01)I;G06K9/62(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í)施例公開了一種張量切分方法、裝置、芯片及介質(zhì)。該方法包括:獲取初始張量和切分參數(shù);根據(jù)所述初始張量和所述切分參數(shù)對(duì)待切分維度上的數(shù)據(jù)進(jìn)行重排,得到重排張量,使得所述待切分維度上的待保留數(shù)據(jù)相鄰排布;根據(jù)所述切分參數(shù)對(duì)所述重排張量進(jìn)行切分,得到結(jié)果張量。上述技術(shù)方通過不支持跨步切分的芯片即可實(shí)現(xiàn)張量的跨步切分,而且減少了寄存器配置數(shù)量。?? |
