[發明專利]矩陣乘法運算方法、裝置及計算機可讀存儲介質

在審

• 申請號: 201910659020.2
• 申請日: 2019-07-19
• 公開（公告）號: CN110377876A
• 公開（公告）日: 2019-10-25
• 發明（設計）人: 劉玉佳;曹建業
• 申請（專利權）人: 廣東省新一代通信與網絡創新研究院
• 主分類號: G06F17/16
• 分類號: G06F17/16
• 代理公司: 深圳市世紀恒程知識產權代理事務所 44287
• 代理人: 張志江

本發明公開了一種矩陣乘法運算方法、裝置及計算機可讀存儲介質，該方法包括：在接收到矩陣乘法運算指令時，獲取待運算第一矩陣的第一標量參數和待運算第二矩陣的第二標量參數；基于第一標量參數和第二標量參數，確定第一矩陣和第二矩陣，并確定第一分割參數和第一填充參數、第二分割參數和第二填充參數；基于第一分割參數和第一填充參數對第一矩陣進行子矩陣分割填充操作得到第一子矩陣序列、基于第二分割參數和所述第二填充參數對第二矩陣進行子矩陣分割填充操作得到第二子矩陣序列；依次選擇第一子矩陣序列中的子矩陣和第二子矩陣序列中的子矩陣做乘法運算得到多個乘積，累加多個乘積得到運算結果。本發明能夠提高矩陣運算速度，減少內存消耗。

技術領域

本發明涉及矩陣運算領域，尤其涉及一種矩陣乘法運算方法、裝置及計算機可讀存儲介質。

背景技術

當前，在計算機領域中，大數據、云確定、人工智能、物聯網、自動駕駛等技術不斷地興起和發展，這些技術中涉及的算法包含各種各樣的矩陣運算，尤其是大矩陣的乘法運算。

現有一般采用通用處理器執行對矩陣的乘法運算，但通用處理器運算能力差，對大矩陣進行乘法運算時非常耗時，運算速度慢，無法滿足這些技術的實時性要求。

發明內容

本發明的主要目的在于提供一種矩陣乘法運算方法、裝置及計算機可讀存儲介質，旨在解決現有矩陣乘法運算方式運算速度慢的技術問題。

為實現上述目的，本發明提供一種矩陣乘法運算方法，所述方法包括以下步驟：

在接收到矩陣乘法運算指令時，獲取待運算第一矩陣的第一標量參數和待運算第二矩陣的第二標量參數；

基于所述第一標量參數和第二標量參數，確定所述第一矩陣和所述第二矩陣，并確定所述第一矩陣的第一分割參數和第一填充參數，以及，確定所述第二矩陣的第二分割參數和第二填充參數；

基于所述第一分割參數和所述第一填充參數，對所述第一矩陣進行子矩陣分割填充操作得到第一子矩陣序列，以及，基于所述第二分割參數和所述第二填充參數，對所述第二矩陣進行子矩陣分割填充操作得到第二子矩陣序列；

依次選擇所述第一子矩陣序列中的子矩陣和所述第二子矩陣序列中的子矩陣做乘法運算，得到多個乘積，并將得到的所述多個乘積進行累加，得到所述第一矩陣和所述第二矩陣的運算結果。

可選地，所述基于所述第一標量參數和第二標量參數，確定所述第一矩陣和所述第二矩陣，并確定所述第一矩陣的第一分割參數和第一填充參數，以及，確定所述第二矩陣的第二分割參數和第二填充參數的步驟包括：

根據所述第一標量參數中所述第一矩陣的首元素地址查找所述第一矩陣，以及，根據所述第二標量參數中所述第二矩陣的首元素地址查找所述第二矩陣；

基于所述第一標量參數中所述第一矩陣的行列數，確定所述第一矩陣的第一分割參數和第一填充參數，以及，基于所述第二標量參數中所述第二矩陣的行列數，確定所述第二矩陣的第二分割參數和第二填充參數。

可選地，所述基于所述第一標量參數中所述第一矩陣的行列數，確定所述第一矩陣的第一分割參數和第一填充參數，以及，基于所述第二標量參數中所述第二矩陣的行列數，確定所述第二矩陣的第二分割參數和第二填充參數的步驟包括：

根據所述第一矩陣的行列數，結合預設單個目標子矩陣行列數，確定所述第一矩陣的第一分割參數和第一填充參數，其中，所述第一分割參數包括所述第一矩陣的分割份數，所述第一填充參數包括所述第一矩陣的待填充行列和待填充元素；以及，

根據所述第二矩陣的行列數，結合預設單個目標子矩陣行列數，確定所述第一矩陣的第二分割參數和第二填充參數，其中，所述第二分割參數包括所述第二矩陣的分割份數，所述第二填充參數包括所述第二矩陣的待填充行列和待填充元素。