[發明專利]人工智能處理器處理二維復數矩陣的方法和設備

在審

• 申請號: 201911349747.7
• 申請日: 2019-12-24
• 公開（公告）號: CN111143766A
• 公開（公告）日: 2020-05-12
• 發明（設計）人: 不公告發明人
• 申請（專利權）人: 上海寒武紀信息科技有限公司
• 主分類號: G06F17/16
• 分類號: G06F17/16;G06F17/14;G06F7/78
• 代理公司: 北京維昊知識產權代理事務所(普通合伙) 11804
• 代理人: 李波;孫新國

1.一種人工智能處理器處理二維復數矩陣的方法，其中，所述二維復數矩陣的尺寸為N×M；與所述二維復數矩陣對應的行系數矩陣的尺寸為M×M，與所述二維復數矩陣對應的列系數矩陣的尺寸為N×N，所述方法包括：

將所述二維復數矩陣加載至所述人工智能處理器的片上存儲單元的第一存儲區中；

將所述行系數矩陣加載至所述片上存儲單元的第二存儲區中；

將所述列系數矩陣加載至所述片上存儲單元的第三存儲區中；

所述人工智能處理器利用所述片上存儲單元中的二維復數矩陣、所述行系數矩陣和列系數矩陣來進行傅里葉變換，以得到運算結果；以及

所述人工智能處理器將所述運算結果傳送至所述片外存儲單元以進行存儲。

2.根據權利要求1所述的方法，其中，通過所述人工智能處理器將所述行系數矩陣轉換為一維數組以加載至所述第二存儲區；將所述列系數矩陣轉換為一維數組以加載至所述第三存儲區。

3.根據權利要求1或2所述的方法，其中，所述二維復數矩陣包括實部矩陣和虛部矩陣。

4.根據權利要求3所述的方法，其中，所述行系數矩陣包括：

第一行系數矩陣，用于存放實部到實部轉換的第一行系數；

第二行系數矩陣，用于存放實部到虛部轉換的第二行系數；

第三行系數矩陣，用于存放虛部到實部轉換的第三行系數；以及

第四行系數矩陣，用于存放虛部到虛部轉換的第四行系數。

5.根據權利要求3所述的方法，其中，所述列系數矩陣包括：

第一列系數矩陣，用于存放實部到實部轉換的第一列系數；

第二列系數矩陣，用于存放實部到虛部轉換的第二列系數；

第三列系數矩陣，用于存放虛部到實部轉換的第三列系數；以及

第四列系數矩陣，用于存放虛部到虛部轉換的第四列系數。

6.根據權利要求1-5中任意一項所述的方法，其中，通過人工智能處理器對所述二維復數矩陣，所述行系數矩陣和列系數矩陣來進行傅里葉變換，以得到運算結果包括：

將所述二維復數矩陣的每一行與所述行系數矩陣的所有行進行卷積運算，以得到第一中間結果；

將所述第一中間結果的同一行元素相加，以得到第一運算結果；

將第一運算結果進行轉置以得到轉置結果，將所述轉置結果的每一行與所述列系數矩陣的所有行進行卷積運算，以得到第二中間結果；

將所述第二中間結果的同一行元素相加，以得到運算結果。

7.根據權利要求1-6中任意一項所述的方法，其中，所述二維復數矩陣按照1×1×1×(N×M)的多維矩陣方式存儲在所述第一存儲單元中。

8.根據權利要求1-7中任意一項所述的方法，其中，所述傅里葉變換是并行進行的。

9.根據權利要求8所述的方法，其中，通過人工智能處理器對所述二維復數矩陣，所述行系數矩陣和列系數矩陣來進行傅里葉變換包括：

通過多個人工智能處理器分別對每個二維復數矩陣中的不同元素，以及與所述元素相對應的行系數矩陣和列系數中的系數元素進行傅里葉變換。

10.根據權利要求9所述的方法，其中，所述不同元素位于不同行或者不同列。

11.一種電子設備，包括：

一個或多個處理器；以及

存儲器，所述存儲器中存儲有計算機可執行指令，當所述計算機可執行指令由所述一個或多個處理器運行時，使得所述電子設備執行如權利要求1-10中任意一項所述的方法。

12.一種計算機可讀存儲介質，包括計算機可執行指令，當所述計算機可執行指令由一個或多個處理器運行時，執行如權利要求1-10中任意一項所述的方法。