一种基于昇腾AI处理器的矩阵乘算子的处理方法及装置

摘要

本发明公开了一种基于昇腾AI处理器的矩阵乘算子的处理方法及装置。其中，该方法包括：获取矩阵乘算子的第一源操作数和第二源操作数；对第一源操作数进行数据转换处理，得到第一转换结果，以及对第二源操作数进行数据转换处理，得到第二转换结果；对第一转换结果进行分块处理，得到第一分块结果，以及对第二转换结果进行分块处理，得到第二分块结果；采用与人工智能处理器中计算核心数量对应的多个计算核心对第一分块结果与第二分块结果进行矩阵乘运算，得到中间运算结果；对中间运算结果进行数据转换处理，得到矩阵乘算子的输出结果，其中，输出结果的输出格式为二维格式。本发明解决了矩阵乘运算的运算效率低的技术问题。