ScriptFunction — PyTorch 2.7 documentation (original) (raw)

class torch.jit.ScriptFunction¶

Functionally equivalent to a ScriptModule, but represents a single function and does not have any attributes or Parameters.

get_debug_state(self: torch._C.ScriptFunction) → torch._C.GraphExecutorState¶

save(self: torch._C.ScriptFunction, filename: str, _extra_files: dict[str, str] = {}) → None¶

save_to_buffer(self: torch._C.ScriptFunction, _extra_files: dict[str, str] = {}) → bytes¶

• ScriptFunction
  • ScriptFunction
    * ScriptFunction.get_debug_state()
    * ScriptFunction.save()
    * ScriptFunction.save_to_buffer()