一种ONNX模型的优化方法及装置

本发明提供一种ONNX模型的优化方法及装置，属于深度学习模型优化技术领域，其中，ONNX模型的优化方法包括：对ONNX模型进行形状推理，获取ONNX模型中各中间节点的形状；获取进行形状推理后的ONNX模型的第一模型状态；对进行形状推理后的ONNX模型进行模型优化处理，并对进行模型优化处理后的ONNX模型再次进行形状推理；再次获取ONNX模型的第二模型状态，将第二模型状态与第一模型状态进行比较，若两者一致，则完成ONNX模型的优化，若两者不一致，则继续进行模型优化处理和形状推理。本发明中的ONNX模型的优化方法，能够更全面地优化模型，实现更少的算子个数以及更小的模型体积，同时不损失模型的精度。