tensorplay.export.export

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

Capture a callable and return an executable graph program.

Parameters:

• model – an nn.Module or plain callable; child modules are inlined.

• args/kwargs – example inputs binding argument defaults.

• dynamic_shapes – dimension specification per argument (dict, sequence, ShapesCollection, or AdditionalInputs).

• strict – reserved for callers of the strict capture contract; capture validation is identical in both modes.

• preserve_module_call_signature – submodule paths whose call metadata is recorded in module_call_graph for module-level tooling.