天眼查显示,上海壁仞科技股份有限公司“人工智能算子中的广播运算方法、装置、设备及介质”专利公布,申请公布日为2024年10月29日,申请公布号为CN118860693A。
本发明提供一种人工智能算子中的广播运算方法、装置、设备及介质,其中人工智能算子中的广播运算方法,包括:响应于广播运算指令,获取广播算子中不同张量分别在每个维度下的步长参数以及总维度;各步长参数和总维度由主控处理器提供;基于各步长参数和总维度分别进行偏移量运算,得到不同张量中第一张量和第二张量各自对应的目标偏移量;基于各目标偏移量分别获取第一张量中的第一元素和第二张量中的第二元素,并对第一元素和第二元素进行预设运算,将运算结果保存至不同张量中的第三张量中。本发明结合前端+后端这一异构硬件框架的强并行性,大幅提高了广播运算方法的通用性和广播效率。