export¶

dragon.onnx.export(
  inputs,
  outputs,
  f,
  input_names=None,
  output_names=None,
  input_shapes=None,
  opset_version=None,
  verbose=False,
  enable_onnx_checker=True
)[source]¶

Export the recorded graph to an onnx model.

Enter into the record mode to export operators into an onnx model:

x = dragon.constant([1, 2, 3])
with dragon.onnx.record():
    y = x * x
dragon.onnx.export(inputs=[x], outputs=[y], f='model.onnx')

Parameters:

• inputs (Union[Sequence, Dict]) – The model inputs.
• outputs (Union[Sequence, Dict]) – The model outputs.
• f (str) – The filename for exporting model.
• input_names (Sequence[str], optional) – The name to the inputs.
• output_names (Sequence[str], optional) – The name to the outputs.
• input_shapes (Union[Sequence, Dict], optional) – The optional rewritten for input shapes.
• opset_version (int, optional) – The version of operator set.
• verbose (bool, optional, default=False) – Whether to print the debug string of graph.
• enable_onnx_checker (bool, optional, default=True) – Whether to check if model is valid.