torch
    Preparing search index...

    Function export_

    • export_(module: Module, sampleInputs: Tensor[]): ExportedProgram

      Export a module's forward pass as an ExportedProgram.

      This traces the module's forward() with the given sample inputs and captures the computation graph. Similar to PyTorch's torch.export.export().

      Named export_ to avoid conflict with the JavaScript export keyword.

      Parameters

      • module: Module

        The nn.Module to export

      • sampleInputs: Tensor[]

        Sample input tensors for tracing

      Returns ExportedProgram

      An ExportedProgram containing the traced graph

    Settings

    Member Visibility