[发明专利]神经网络模型的SDK输出方法、装置、电子设备及存储介质

权利要求书

1.一种神经网络模型的SDK输出方法，其特征在于，包括：

获取待生成软件开发工具包SDK的神经网络模型的模型描述信息，以及所述待生成SDK对应的硬件平台类型和编程语言类型，所述神经网络模型的模型描述信息包括神经网络模型的模型参数，所述神经网络模型的模型参数包括超参数和训练参数；

根据所述硬件平台类型和编程语言类型，从预先生成的SDK输出模板中匹配与所述硬件平台类型和编程语言类型对应的SDK输出模板；

将所述模型描述信息输入到匹配到的SDK输出模板中，以替换所述SDK输出模板中神经网络模型相关结构部分，得到所述神经网络模型在所述硬件平台类型和编程语言类型下的SDK，所述神经网络模型相关结构部分包括所述神经网络模型的结构实现部分。

2.根据权利要求1所述的方法，其特征在于，在所述根据所述硬件平台类型和编程语言类型，从预先生成的SDK输出模板中匹配与所述硬件平台类型和编程语言类型对应的SDK输出模板之前，还包括：

获取所述神经网络模型的模型类型；

所述根据所述硬件平台类型和编程语言类型，从预先生成的SDK输出模板中匹配与所述硬件平台类型和编程语言类型对应的SDK输出模板，包括：

根据所述模型类型、所述硬件平台类型和编程语言类型，从预先生成的SDK输出模板中匹配与所述模型类型、所述硬件平台类型和编程语言类型对应的SDK输出模板。

3.根据权利要求1所述的方法，其特征在于，根据所述硬件平台类型和编程语言类型，从预先生成的SDK输出模板中匹配与所述硬件平台类型和编程语言类型对应的SDK输出模板，包括：

根据所述硬件平台类型和编程语言类型，与预先生成的SDK输出模板对应的模板元描述信息进行匹配；

当所述硬件平台类型和编程语言类型与所述模板元描述信息匹配时，确定该模板元描述信息对应的SDK输出模板为与所述硬件平台类型和编程语言类型对应的SDK输出模板。

4.根据权利要求1所述的方法，其特征在于，将所述模型描述信息输入到匹配到的SDK输出模板中，以替换所述SDK输出模板中神经网络模型相关结构部分，得到所述神经网络模型在所述硬件平台类型和编程语言类型下的SDK，包括：

将所述模型描述信息输入到匹配到的SDK输出模板中，以替换所述SDK输出模板中神经网络模型相关结构部分；

调用所述硬件平台类型对应的公共库对所述神经网络模型进行调优，并调用所述硬件平台类型对应的针对硬件指令集的加速进行硬件加速，得到所述神经网络模型在所述硬件平台类型和编程语言类型下的SDK。

5.根据权利要求1所述的方法，其特征在于，在得到所述神经网络模型在所述硬件平台类型和编程语言类型下的SDK之后，还包括：

对所述SDK进行验证，以生成验证结果。

6.一种神经网络模型的SDK输出装置，其特征在于，包括：

信息获取模块，用于获取待生成SDK的神经网络模型的模型描述信息，以及所述待生成SDK对应的硬件平台类型和编程语言类型，所述神经网络模型的模型描述信息包括神经网络模型的模型参数，所述神经网络模型的模型参数包括超参数和训练参数；

模板匹配模块，用于根据所述硬件平台类型和编程语言类型，从预先生成的SDK输出模板中匹配与所述硬件平台类型和编程语言类型对应的SDK输出模板；

SDK生成模块，用于将所述模型描述信息输入到匹配到的SDK输出模板中，以替换所述SDK输出模板中神经网络模型相关结构部分，得到所述神经网络模型在所述硬件平台类型和编程语言类型下的SDK，所述神经网络模型相关结构部分包括所述神经网络模型的结构实现部分。