首页/ 无锡灵汐类脑科技有限公司/ 专利详情

基于广播机制进行张量计算的方法、装置、芯片及介质

基本信息

申请号 CN202011528418.1 申请日 -
公开（公告）号 CN112597424A 公开（公告）日 2021-04-02
申请公布号 CN112597424A 申请公布日 2021-04-02
分类号 G06F17/10(2006.01)I 分类 计算；推算；计数;
发明人 吴欣洋;李涵;戚海涛;丁瑞强;冯开革 申请（专利权）人 无锡灵汐类脑科技有限公司
代理机构 北京品源专利代理有限公司 代理人 孟金喆
地址 214028 江苏省无锡市新吴区弘毅路10号金乾座1701-1710室
法律状态 -

摘要

摘要 本发明实施例公开了一种基于广播机制进行张量计算的方法、装置、芯片及介质。该方法包括：获取第一张量和第二张量，第一张量的维数大于或等于第二张量的维数；判断第一张量和第二张量是否满足广播机制条件，若满足，则确定与第一张量和第二张量分别对应的第一目标张量和第二目标张量，第一目标张量和第二目标张量的后缘维度的轴长度相符；通过通用计算硬件单元组，采用广播机制对第一目标张量和第二目标张量进行张量计算，得到张量计算备选结果；根据张量计算备选结果，确定与第一张量和第二张量对应的目标计算结果。上述技术方案实现了在无支持广播计算的专用底层硬件时全面支持广播机制中所有形式的广播操作。