在 PyTorch 中使用 numpy 函数并不会影响导出 ONNX 模型的过程。可以使用 PyTorch 的 torch.onnx.export() 函数将模型导出为 ONNX 格式。使用方法如下:
import torch.onnx# Define the model
model = ...
# Input to the model
x = ...
# Export the model
torch.onnx.export(model, x, "model.onnx")
导出模型时,可以通过指定参数 input_names
和 output_names
来命名