[發明專利]多輸入多輸出矩陣平均值池化向量化實現方法

有效

• 申請號: 201711478728.5
• 申請日: 2017-12-29
• 公開（公告）號: CN108205703B
• 公開（公告）日: 2021-01-12
• 發明（設計）人: 郭陽;張軍陽;楊超;田希;扈嘯;李斌
• 申請（專利權）人: 中國人民解放軍國防科技大學
• 主分類號: G06N3/04
• 分類號: G06N3/04
• 代理公司: 湖南兆弘專利事務所(普通合伙) 43008
• 代理人: 周長清

1.一種多輸入多輸出矩陣平均值池化的向量化實現方法，其特征在于，步驟為：

S1：根據向量處理器中向量處理單元VPE的數量M，輸入特征圖的數量N，尺寸為n×n，平均值池化的移動步長為s，池化窗口k，確定向量處理器單核同時計算的輸入特征圖數量；

S2：將M個輸入特征圖按照第三維進行排序；

S3：步驟S2進行N/M次，直至完成所有N張輸入特征圖的排序；

S4：將步驟S3中排序好的輸入特征圖由DMA傳輸至向量處理器核內AM中；

S5：向量加載AM中的第1行，依次與接下來的數據行進行累加，共計累加k×k次，并將累加結果與1/k²相乘，同時得出M個輸入特征圖對應位置池化窗口為k×k的取平均值池化結果；

S6：根據水平移動步長s，順移至下一個池化窗口，進行同步驟S5的計算，得出M個輸入特征圖第2個k×k池化窗口的平均值池化結果；

S7：重復步驟S5-S6直至完成整個M個輸入特征圖所有池化窗口的平均值池化操作；

S8：重復步驟S6-S7N/M次，最終完成N張輸入特征圖的平均值池化操作。

2.根據權利要求1所述的多輸入多輸出矩陣平均值池化的向量化實現方法，其特征在于，所述步驟S1中，輸入特征圖的數量N遠大于向量處理單元VPE的數量M，且N為M的整數倍。

3.根據權利要求1所述的多輸入多輸出矩陣平均值池化的向量化實現方法，其特征在于，所述步驟S1中，輸入特征圖是一個三維矩陣，即高度、寬度、數量。

4.根據權利要求3所述的多輸入多輸出矩陣平均值池化的向量化實現方法，其特征在于，所述步驟S1中的三維矩陣為方陣，即高度和寬度相等。

5.根據權利要求1-4中任意一項所述的多輸入多輸出矩陣平均值池化的向量化實現方法，其特征在于，所述步驟S1中平均值池化的移動步長分為水平移動步長和垂直移動步長。

6.根據權利要求5所述的多輸入多輸出矩陣平均值池化的向量化實現方法，其特征在于，所述步驟S1中取水平移動步長和垂直移動步長相同。

7.根據權利要求1-4中任意一項所述的多輸入多輸出矩陣平均值池化的向量化實現方法，其特征在于，所述步驟S1中的池化尺寸取方陣，即最大值池化窗口為k×k。

8.根據權利要求1-4中任意一項所述的多輸入多輸出矩陣平均值池化的向量化實現方法，其特征在于，所述步驟S1中，若輸入特征圖的數量不為VPE數量的整數倍，則多余的輸入特征圖由若干VPE進行處理。

9.根據權利要求1-4中任意一項所述的多輸入多輸出矩陣平均值池化的向量化實現方法，其特征在于，所述步驟S6中，輸入特征圖每次取一個與k×k對應大小的池化窗口，且該窗口根據移動步長先水平移動，后垂直移動，移動順序為，從左到右，從上到下。