[发明专利]内存管理方法、装置及存储介质

权利要求书

1.一种内存管理方法，其特征在于，应用于终端中，所述方法包括：

根据神经网络中的特征单元连接关系，确定所述神经网络的至少一个分支，所述神经网络包括按照顺序排列的多个网络层，每个网络层包括至少一个特征单元，每个分支由位于不同网络层的多个特征单元连接构成；

对于每个分支，根据所述分支上每个特征单元需要的输出内存的尺寸，为所述分支分配第一内存和第二内存，第一内存尺寸不小于第二内存尺寸，所述第一内存尺寸和所述第二内存尺寸均不小于所述分支需要的其他内存尺寸；

当所述第一内存尺寸与所述第二内存尺寸相等时，将所述第一内存和所述第二内存轮流作为所述分支上特征单元的输入内存和输出内存；

当所述第一内存尺寸大于所述第二内存尺寸时，确定所述分支上需要的输出内存的尺寸为所述第一内存尺寸的指定特征单元，将所述第一内存作为所述指定特征单元的输出内存，按照所述分支上每个特征单元与所述指定特征单元之间的位置关系，将所述第一内存和所述第二内存作为所述分支上除指定特征单元以外的每个特征单元的输入内存或输出内存，以使所述第一内存和所述第二内存轮流作为所述分支上特征单元的输入内存和输出内存。

2.根据权利要求1所述的方法，其特征在于，所述根据所述分支上每个特征单元需要的输出内存的尺寸，为所述分支分配第一内存和第二内存之前，所述方法还包括：

对所述分支上每个特征单元需要输出的数据量进行统计，得到所述分支上每个特征单元需要的输出内存的尺寸。

3.根据权利要求1所述的方法，其特征在于，所述根据所述分支上每个特征单元需要的输出内存的尺寸，为所述分支分配第一内存和第二内存，包括：

按照从大到小的顺序，对所述分支上每个特征单元需要的输出内存的尺寸进行排序；

将排在第一位的尺寸作为第一内存尺寸，将排在第二位的尺寸作为第二内存尺寸；

为所述分支分配与所述第一内存尺寸匹配的第一内存和与所述第二内存尺寸匹配的第二内存。

4.根据权利要求1所述的方法，其特征在于，所述将所述第一内存和所述第二内存轮流作为所述分支上特征单元的输入内存和输出内存，包括：

将所述第一内存作为所述分支上第一特征单元的输出内存，且作为第二特征单元的输入内存，将所述第二内存作为所述第二特征单元的输出内存；

其中，所述第一特征单元为所述分支上的任一特征单元，所述第二特征单元为所述分支上所述第一特征单元的下一个特征单元。

5.根据权利要求1-4任一项所述的方法，其特征在于，所述将所述第一内存和所述第二内存轮流作为所述分支上特征单元的输入内存和输出内存之后，所述方法还包括：

在基于所述分支进行计算的过程中，将第一特征单元输出的数据存储于所述第一特征单元的输出内存中；

从所述第一特征单元的输出内存中提取所述第一特征单元输出的数据，通过第二特征单元对提取的数据进行计算，将所述第二特征单元输出的数据存储于所述第二特征单元的输出内存中；

其中，所述第一特征单元为所述分支上的任一特征单元，所述第二特征单元为所述分支上所述第一特征单元的下一个特征单元，所述第一特征单元的输出内存与所述第二特征单元的输入内存相同。