tensorplay.onnx

Functions

export() [source]

export(model, args, f, input_names=None, output_names=None, opset_version=13)

Export a model to ONNX format.

Args

• model: callable (nn.Module or function)
• args: tuple of input tensors
• f: file path or file-like object
• input_names: list of input names (optional)
• output_names: list of output names (optional)
• opset_version: int (default 13)