[發明專利]數據處理方法及裝置

有效

• 申請號: 201910197813.7
• 申請日: 2019-03-15
• 公開（公告）號: CN111695682B
• 公開（公告）日: 2022-11-01
• 發明（設計）人: 不公告發明人
• 申請（專利權）人: 上海寒武紀信息科技有限公司
• 主分類號: G06N3/063
• 分類號: G06N3/063
• 代理公司: 北京林達劉知識產權代理事務所(普通合伙) 11277
• 代理人: 劉新宇

1.一種數據處理方法，其特征在于，所述方法包括：

在輸入數據的第一數據格式與第一處理器支持的第二數據格式不同時，根據第一處理器的運算單元的個數和/或預設維度數，將輸入數據由第一數據格式轉換為第二數據格式，

其中，所述輸入數據為張量數據，第一數據格式和第二數據格式的數據擺放順序不同，預設維度數為第一處理器的運算單元一次讀取的數據個數；

所述根據第一處理器的運算單元的個數和/或預設維度數，將輸入數據由第一數據格式轉換為第二數據格式，包括：

對于第二數據格式的多個第二維度中的最低維度，根據輸入數據的參數信息確定第一數據格式的多個第一維度中與所述最低維度對應的第三維度；其中，最低維度為讀寫數據時首先被讀寫的維度方向，所述參數信息包括用于描述第一數據格式的擺放順序的信息；

根據第三維度的維度數、預設維度數以及所述個數對輸入數據在第三維度方向進行補齊，并根據預設維度數將補齊后的輸入數據在第三維度方向上進行分段；

將分段后的輸入數據由第一數據格式轉換為第二數據格式。

2.根據權利要求1所述的方法，其特征在于，根據第三維度的維度數、預設維度數以及所述個數對輸入數據在第三維度方向進行補齊，包括：

若第三維度的維度數小于預設維度數，則對輸入數據在第三維度方向補齊到預設維度數。

3.根據權利要求2所述的方法，其特征在于，根據第三維度的維度數、預設維度數以及所述個數對輸入數據在第三維度方向進行補齊，還包括：

若第三維度的維度數大于預設維度數，則計算第三維度的維度數與預設維度數的余數，若所述余數與所述預設維度數滿足第一預設條件，則需要補齊的維度數與所述余數的和為所述個數的整數倍，若所述余數與所述預設維度數不滿足所述第一預設條件，則需要補齊的維度數與所述余數的和為所述預設維度數。

4.根據權利要求1-3任意一項所述的方法，其特征在于，所述第二數據格式為NHWC格式，其中，N代表數量、C代表通道數、H代表高度、W代表寬度，NHWC格式中C為最低維度。

5.根據權利要求4所述的方法，其特征在于，將分段后的輸入數據由第一數據格式轉換為第二數據格式，包括：

在N維度方向上針對每一段分段后的數據執行以下操作：

針對多個第二維度中的HWC維度，按照CWH的順序從該段輸入數據中與第二維度對應的第一維度方向讀取輸入數據，并按照讀取的順序存儲得到第二數據格式的輸入數據。

6.根據權利要求1-3任意一項所述的方法，其特征在于，所述方法應用于第一處理器，所述第一處理器為人工智能處理器，

在所述第一處理器中神經網絡的輸入端和神經網絡的第一層算子之間設有第一算子，所述第一算子用于根據第一處理器的運算單元的個數和/或預設維度數，將輸入數據由第一數據格式轉換為第二數據格式。

7.根據權利要求6所述的方法，其特征在于，在所述神經網絡的最后一層算子和輸出端之間設有第二算子，所述第二算子用于根據第一處理器的運算單元的個數和/或預設維度數，將神經網絡的輸出數據由第二數據格式轉換為第一數據格式。