[发明专利]一种模型文件的生成方法、装置、电子设备及存储介质

权利要求书

1.一种模型文件的生成方法，其特征在于，所述生成方法包括：

获取目标模型对应的模型函数；其中，所述模型函数包括多个算子；

将所述模型函数中的每个算子转换成第一通用表达式；其中，所述第一通用表达式包括该算子的多个算子属性；

根据每个算子对应的输入信息和输出信息，确定所述多个算子中任意两个算子之间的关联关系；

根据每个算子对应的第一通用表达式，以及所述多个算子中任意两个算子之间的关联关系，生成所述目标模型对应的模型文件。

2.根据权利要求1所述的模型文件的生成方法，其特征在于，所述将所述模型函数中的每个算子转换成第一通用表达式，包括：

针对所述模型函数中的每个算子，根据该算子所属的类别，确定该算子对应的多个算子属性，将该算子转换为包括所述多个算子属性的第一通用表达式。

3.根据权利要求2所述的模型文件的生成方法，其特征在于，通过如下方式确定每个算子所属的类别：

查找所述模型函数中每个算子的存储地址；

基于所述存储地址的查找结果，将与所述查找结果相匹配的类别，确定为该算子所属的类别。

4.根据权利要求1所述的模型文件的生成方法，其特征在于，所述根据每个算子对应的第一通用表达式，以及所述多个算子中任意两个算子之间的关联关系，生成所述目标模型对应的模型文件，包括：

基于预设的多个模型属性，生成所述模型函数对应的包括所述多个模型属性的第二通用表达式；

根据每个算子对应的第一通用表达式，所述模型函数对应的第二通用表达式，以及所述多个算子中任意两个算子之间的关联关系，生成所述目标模型对应的模型文件。

5.根据权利要求2所述的模型文件的生成方法，其特征在于，所述算子所属的类别包括：自定义算子、框架语义算子。

6.根据权利要求1所述的模型文件的生成方法，其特征在于，所述根据每个算子对应的第一通用表达式，以及所述多个算子中任意两个算子之间的关联关系，生成所述目标模型对应的模型文件，包括：

根据所述多个算子中任意两个算子之间的关联关系，确定所述多个算子对应的组合方式；

根据每个算子对应的第一通用表达式，以及所述多个算子对应的组合方式，生成所述目标模型对应的模型文件。

7.根据权利要求6所述的模型文件的生成方法，其特征在于，所述多个算子对应的组合方式包括：顺序执行、一对多执行、多对多执行。

8.一种模型文件的生成装置，其特征在于，所述生成装置包括：

获取模块，用于获取目标模型对应的模型函数；其中，所述模型函数包括多个算子；

转换模块，用于将所述模型函数中的每个算子转换成第一通用表达式；其中，所述第一通用表达式包括该算子的多个算子属性；

第一确定模块，用于根据每个算子对应的输入信息和输出信息，确定所述多个算子中任意两个算子之间的关联关系；

生成模块，用于根据每个算子对应的第一通用表达式，以及所述多个算子中任意两个算子之间的关联关系，生成所述目标模型对应的模型文件。

9.一种电子设备，其特征在于，包括：处理器、存储器和总线，所述存储器存储有所述处理器可执行的机器可读指令，当电子设备运行时，所述处理器与所述存储器之间通过总线通信，所述处理器执行所述机器可读指令，以执行如权利要求1至7任一项所述的模型文件的生成方法的步骤。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有计算机程序，该计算机程序被处理器运行时执行如权利要求1至7任一项所述的模型文件的生成方法的步骤。