tensorplay.export.export_for_training

tensorplay.export.export_for_training(model: Callable[[...], Any], *args: Any, dynamic_shapes: Any = None, **kwargs: Any) → ExportedProgram[source]

Capture a callable while retaining its mutable training state.