张量切分方法、装置、芯片及介质
基本信息
申请号 | CN202011479734.4 | 申请日 | - |
公开(公告)号 | CN112507173A | 公开(公告)日 | 2021-03-16 |
申请公布号 | CN112507173A | 申请公布日 | 2021-03-16 |
分类号 | G06F16/901(2019.01)I;G06K9/62(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 吴欣洋;李涵;戚海涛;丁瑞强;冯开革 | 申请(专利权)人 | 无锡灵汐类脑科技有限公司 |
代理机构 | 北京品源专利代理有限公司 | 代理人 | 孟金喆 |
地址 | 214028 江苏省无锡市新吴区弘毅路10号金乾座1701-1710室 | ||
法律状态 | - |
摘要
摘要 | 本发明实施例公开了一种张量切分方法、装置、芯片及介质。该方法包括:获取初始张量和切分参数;根据所述初始张量和所述切分参数对待切分维度上的数据进行重排,得到重排张量,使得所述待切分维度上的待保留数据相邻排布;根据所述切分参数对所述重排张量进行切分,得到结果张量。上述技术方通过不支持跨步切分的芯片即可实现张量的跨步切分,而且减少了寄存器配置数量。 |
