[發明專利]數據處理方法、裝置、計算機設備和存儲介質

在審

• 申請號: 201911061027.0
• 申請日: 2019-11-01
• 公開（公告）號: CN112765538A
• 公開（公告）日: 2021-05-07
• 發明（設計）人: 不公告發明人
• 申請（專利權）人: 中科寒武紀科技股份有限公司
• 主分類號: G06F17/15
• 分類號: G06F17/15;G06N3/04;G06N3/063
• 代理公司: 北京林達劉知識產權代理事務所(普通合伙) 11277
• 代理人: 劉新宇

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

將第一卷積核按照步長N進行拆分，得到多個第二卷積核；

將第一輸入數據按照步長N進行拆分，得到與多個所述第一卷積核對應的多個第二輸入數據；

針對任一所述第二輸入數據，將該第二輸入數據與對應的所述第二卷積核執行winograd卷積操作，得到所述第二輸入數據對應的卷積結果；

確定所述多個第二輸入數據對應的卷積結果的和為所述第一卷積核與所述第一輸入數據的卷積結果。

2.根據權利要求1所述的方法，其特征在于，所述第二輸入數據與所述第二卷積核的對應關系，具體為：所述第二輸入數據中的第一個元素在所述第一輸入數據中的位置與所述第二卷積核中第一個元素在所述第一卷積核中的位置相同。

3.根據權利要求1或2所述的方法，其特征在于，所述將第一卷積核按照步長N進行拆分，得到多個第二卷積核，包括：

針對第一卷積核的行與列，采用間隔為N-1步對第一卷積核進行拆分，得到多個第二卷積核；

所述將第一輸入數據按照步長N進行拆分，得到與多個所述第一卷積核對應的多個第二輸入數據，包括：

針對第一輸入數據的行與列，采用間隔為N-1步對第一輸入數據進行拆分，得到與多個所述第一卷積核對應的多個第二輸入數據。

4.根據權利要求3所述的方法，其特征在于，所述針對第一卷積核的行與列，采用間隔為N-1步對第一卷積核進行拆分，得到多個第二卷積核，包括：

遍歷第一卷積核中的元素，重復每間隔N-1行確定一行為目標行，針對所述目標行，每間隔N-1列獲取一個元素，獲取的多個元素組成一個第二卷積核的過程，直到遍歷完第一卷積核中的元素；

針對第一輸入數據的行與列，采用間隔為N-1步對第一輸入數據進行拆分，得到與多個所述第一卷積核對應的多個第二輸入數據，包括：

遍歷第一輸入數據中的元素，重復每間隔N-1行確定一行為目標行，針對所述目標行，每間隔N-1列獲取一個元素，獲取的多個元素組成一個第二輸入數據的過程，直到遍歷完第一輸入數據中的元素。

5.根據權利要求1至4任一項所述的方法，其特征在于，所述針對任一所述第二輸入數據，將該第二輸入數據與對應的所述第二卷積核執行winograd卷積操作，得到所述第二輸入數據對應的卷積結果，包括：

將所述第二輸入數據的winograd正變換拆解為求和運算，進行計算得到所述第二輸入數據的winograd正變換結果；

將所述第二卷積核的winograd正變換拆解為求和運算，進行計算得到第二卷積核的winograd正變換結果；

執行所述第二輸入數據的winograd正變換結果與所述第二卷積核的winograd正變換結果的對位乘操作，得到對位乘結果；

將對所述對位乘結果的winograd逆變換拆解為求和運算，得到所述第二輸入數據與對應的所述第二卷積核的卷積結果。

6.根據權利要求5所述的方法，其特征在于，所述將所述第二輸入數據的winograd正變換拆解為求和運算，進行計算得到所述第二輸入數據的winograd正變換結果，包括：

將所述第二輸入數據拆解為多個第一子張量，對所述多個第一子張量進行winograd正變換并求和得到所述第二輸入數據的winograd正變換結果。

7.根據權利要求5所述的方法，其特征在于，所述將所述第二卷積核的winograd正變換拆解為求和運算，進行計算得到第二卷積核的winograd正變換結果，包括：

將所述第二卷積核拆解為多個第二子張量，對所述多個第二子張量進行winograd正變換并求和得到所述第二卷積核的winograd正變換結果。

8.根據權利要求6所述的方法，其特征在于，所述多個第一子張量的個數與所述第二輸入數據的不為0元素的個數相同，所述多個第一子張量中的每個第一子張量中有一個元素與所述第二輸入數據中的對應位置的元素相同、其他元素均為0。