pytorch_pfn_extras.onnx.pfto_exporter.export.export#

pytorch_pfn_extras.onnx.pfto_exporter.export.export(model, args, f, *, chrome_tracing='', **kwargs)#
Parameters:

  • model (Callable) –

  • args (Sequence[Any]) –

  • f (Union[str, IO]) –

  • chrome_tracing (str) –

  • kwargs (object) –

Return type:

Any