[發(fā)明專利]一種數(shù)據(jù)處理方法、裝置、芯片、設(shè)備及介質(zhì)

在審

• 申請?zhí)枺?/td>: 202111282230.8
• 申請日: 2021-11-01
• 公開（公告）號: CN114037054A
• 公開（公告）日: 2022-02-11
• 發(fā)明（設(shè)計）人: 張靜;周全
• 申請（專利權(quán)）人: 青島信芯微電子科技股份有限公司
• 主分類號: G06N3/04
• 分類號: G06N3/04;G06N3/063;G06F12/02;G06F15/78
• 代理公司: 北京同達信恒知識產(chǎn)權(quán)代理有限公司 11291
• 代理人: 杜晶

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

根據(jù)內(nèi)存的存儲空間和神經(jīng)網(wǎng)絡(luò)層中卷積核的第一尺寸，確定輸出數(shù)據(jù)的第二尺寸和卷積核的數(shù)量；

根據(jù)所述輸出數(shù)據(jù)的第二尺寸，確定輸入數(shù)據(jù)的第三尺寸；

根據(jù)所述輸入數(shù)據(jù)的第三尺寸和所述卷積核的數(shù)量，對外存中的數(shù)據(jù)進行切分，將切分后的數(shù)據(jù)輸入到內(nèi)存中。

2.如權(quán)利要求1所述的方法，其特征在于，所述根據(jù)內(nèi)存的存儲空間和神經(jīng)網(wǎng)絡(luò)層中卷積核的第一尺寸，確定輸出數(shù)據(jù)的第二尺寸和卷積核的數(shù)量，包括：

根據(jù)所述卷積核的第一尺寸、所述卷積核的第一通道數(shù)和一組卷積核的數(shù)量，確定所述一組卷積核對應(yīng)的第一數(shù)據(jù)量；

根據(jù)輸出數(shù)據(jù)的初始尺寸，確定輸入數(shù)據(jù)的初始尺寸；

根據(jù)所述第一數(shù)據(jù)量、所述輸出數(shù)據(jù)的初始尺寸和所述輸入數(shù)據(jù)的初始尺寸，確定一次計算對應(yīng)的第二數(shù)據(jù)量；

如果所述第二數(shù)據(jù)量超過所述內(nèi)存的存儲空間，和/或如果所述輸入數(shù)據(jù)的初始尺寸大于或等于設(shè)定尺寸，對所述輸出數(shù)據(jù)的初始尺寸進行調(diào)整，直至調(diào)整后的第二數(shù)據(jù)量小于或等于所述內(nèi)存的存儲空間，且調(diào)整后的輸入數(shù)據(jù)的初始尺寸小于所述設(shè)定尺寸；

如果所述第二數(shù)據(jù)量超過所述內(nèi)存的存儲空間，且所述輸入數(shù)據(jù)的初始尺寸小于所述設(shè)定尺寸，將所述輸出數(shù)據(jù)的初始尺寸確定為所述輸出數(shù)據(jù)的第二尺寸。

3.如權(quán)利要求2所述的方法，其特征在于，所述根據(jù)內(nèi)存的存儲空間和神經(jīng)網(wǎng)絡(luò)層中卷積核的第一尺寸，確定卷積核的數(shù)量，包括：

根據(jù)所述內(nèi)存的存儲空間、所述輸出數(shù)據(jù)的第二尺寸和所述一組卷積核對應(yīng)的第一數(shù)據(jù)量，確定卷積核的組數(shù)，并確定所述組數(shù)的卷積核的數(shù)量。

4.如權(quán)利要求1-3任一項所述的方法，其特征在于，所述對外存中的數(shù)據(jù)進行切分，包括：

對所述外存中的數(shù)據(jù)進行重新排布，對重新排布后的數(shù)據(jù)進行切分。

5.如權(quán)利要求4所述的方法，其特征在于，所述數(shù)據(jù)包括特征數(shù)據(jù)，所述對所述外存中的數(shù)據(jù)進行重新排布，包括：

按照所述特征數(shù)據(jù)的輸入順序進行排序，按照第一設(shè)定數(shù)量對排序后的所述特征數(shù)據(jù)進行分組；

在每組特征數(shù)據(jù)中，分別按照所述特征數(shù)據(jù)的輸入順序、像素點的排布順序，對該組特征數(shù)據(jù)中的特征數(shù)據(jù)中的像素點進行重新排布。

6.如權(quán)利要求4所述的方法，其特征在于，所述數(shù)據(jù)包括卷積核數(shù)據(jù)，所述對所述外存中的數(shù)據(jù)進行重新排布，包括：

按照所述卷積核數(shù)據(jù)的輸入順序進行排序，按照第二設(shè)定數(shù)量對排序后的所述卷積核數(shù)據(jù)進行分組；

在每組卷積核數(shù)據(jù)中，分別按照所述卷積核數(shù)據(jù)的輸入順序、通道的排布順序、像素點的排布順序，對該組卷積核數(shù)據(jù)中的卷積核數(shù)據(jù)中的像素點進行重新排布。

7.一種數(shù)據(jù)處理裝置，其特征在于，所述裝置包括：

第一確定模塊，用于根據(jù)內(nèi)存的存儲空間和神經(jīng)網(wǎng)絡(luò)層中卷積核的第一尺寸，確定輸出數(shù)據(jù)的第二尺寸和卷積核的數(shù)量；

第二確定模塊，用于根據(jù)所述輸出數(shù)據(jù)的第二尺寸，確定輸入數(shù)據(jù)的第三尺寸；

切分模塊，用于根據(jù)所述輸入數(shù)據(jù)的第三尺寸和所述卷積核的數(shù)量，對外存中的數(shù)據(jù)進行切分，將切分后的數(shù)據(jù)輸入到內(nèi)存中。

8.一種芯片，其特征在于，所述芯片用于執(zhí)行權(quán)利要求1-6中任一所述數(shù)據(jù)處理方法的步驟。

9.一種電子設(shè)備，其特征在于，所述電子設(shè)備至少包括處理器和存儲器，所述處理器用于執(zhí)行存儲器中存儲的計算機程序時實現(xiàn)根據(jù)權(quán)利要求1-6中任一所述數(shù)據(jù)處理方法的步驟。

10.一種計算機可讀存儲介質(zhì)，其特征在于，其存儲有計算機程序，所述計算機程序被處理器執(zhí)行時實現(xiàn)根據(jù)權(quán)利要求1-6中任一所述數(shù)據(jù)處理方法的步驟。