[发明专利]矩阵乘法运算方法、装置及计算机可读存储介质

说明书

本发明公开了一种矩阵乘法运算方法、装置及计算机可读存储介质，该方法包括：在接收到矩阵乘法运算指令时，获取待运算第一矩阵的第一标量参数和待运算第二矩阵的第二标量参数；基于第一标量参数和第二标量参数，确定第一矩阵和第二矩阵，并确定第一分割参数和第一填充参数、第二分割参数和第二填充参数；基于第一分割参数和第一填充参数对第一矩阵进行子矩阵分割填充操作得到第一子矩阵序列、基于第二分割参数和所述第二填充参数对第二矩阵进行子矩阵分割填充操作得到第二子矩阵序列；依次选择第一子矩阵序列中的子矩阵和第二子矩阵序列中的子矩阵做乘法运算得到多个乘积，累加多个乘积得到运算结果。本发明能够提高矩阵运算速度，减少内存消耗。

技术领域

本发明涉及矩阵运算领域，尤其涉及一种矩阵乘法运算方法、装置及计算机可读存储介质。

背景技术

当前，在计算机领域中，大数据、云确定、人工智能、物联网、自动驾驶等技术不断地兴起和发展，这些技术中涉及的算法包含各种各样的矩阵运算，尤其是大矩阵的乘法运算。

现有一般采用通用处理器执行对矩阵的乘法运算，但通用处理器运算能力差，对大矩阵进行乘法运算时非常耗时，运算速度慢，无法满足这些技术的实时性要求。

发明内容

本发明的主要目的在于提供一种矩阵乘法运算方法、装置及计算机可读存储介质，旨在解决现有矩阵乘法运算方式运算速度慢的技术问题。

为实现上述目的，本发明提供一种矩阵乘法运算方法，所述方法包括以下步骤：

在接收到矩阵乘法运算指令时，获取待运算第一矩阵的第一标量参数和待运算第二矩阵的第二标量参数；

基于所述第一标量参数和第二标量参数，确定所述第一矩阵和所述第二矩阵，并确定所述第一矩阵的第一分割参数和第一填充参数，以及，确定所述第二矩阵的第二分割参数和第二填充参数；

基于所述第一分割参数和所述第一填充参数，对所述第一矩阵进行子矩阵分割填充操作得到第一子矩阵序列，以及，基于所述第二分割参数和所述第二填充参数，对所述第二矩阵进行子矩阵分割填充操作得到第二子矩阵序列；

依次选择所述第一子矩阵序列中的子矩阵和所述第二子矩阵序列中的子矩阵做乘法运算，得到多个乘积，并将得到的所述多个乘积进行累加，得到所述第一矩阵和所述第二矩阵的运算结果。

可选地，所述基于所述第一标量参数和第二标量参数，确定所述第一矩阵和所述第二矩阵，并确定所述第一矩阵的第一分割参数和第一填充参数，以及，确定所述第二矩阵的第二分割参数和第二填充参数的步骤包括：

根据所述第一标量参数中所述第一矩阵的首元素地址查找所述第一矩阵，以及，根据所述第二标量参数中所述第二矩阵的首元素地址查找所述第二矩阵；

基于所述第一标量参数中所述第一矩阵的行列数，确定所述第一矩阵的第一分割参数和第一填充参数，以及，基于所述第二标量参数中所述第二矩阵的行列数，确定所述第二矩阵的第二分割参数和第二填充参数。

可选地，所述基于所述第一标量参数中所述第一矩阵的行列数，确定所述第一矩阵的第一分割参数和第一填充参数，以及，基于所述第二标量参数中所述第二矩阵的行列数，确定所述第二矩阵的第二分割参数和第二填充参数的步骤包括：

根据所述第一矩阵的行列数，结合预设单个目标子矩阵行列数，确定所述第一矩阵的第一分割参数和第一填充参数，其中，所述第一分割参数包括所述第一矩阵的分割份数，所述第一填充参数包括所述第一矩阵的待填充行列和待填充元素；以及，

根据所述第二矩阵的行列数，结合预设单个目标子矩阵行列数，确定所述第一矩阵的第二分割参数和第二填充参数，其中，所述第二分割参数包括所述第二矩阵的分割份数，所述第二填充参数包括所述第二矩阵的待填充行列和待填充元素。