TorchScript

创建日期: 2018 年 9 月 7 日 | 最后更新日期: 2025 年 7 月 16 日

警告

TorchScript 已弃用，请使用 torch.export 代替。

创建 TorchScript 代码

script	脚本化函数。
trace	跟踪函数并返回一个可执行函数或 ScriptFunction，它将通过即时编译进行优化。
script_if_tracing	在跟踪期间首次调用 fn 时进行编译。
trace_module	跟踪模块并返回一个可执行 ScriptModule，它将通过即时编译进行优化。
fork	创建一个执行 func 的异步任务，并返回该执行结果的引用。
wait	强制完成 torch.jit.Future[T] 异步任务，并返回任务的结果。
ScriptModule	C++ torch::jit::Module 的包装器，包含方法、属性和参数。
ScriptFunction	功能上等同于 ScriptModule，但它代表单个函数，没有任何属性或参数。
freeze	冻结 ScriptModule，将子模块和属性内联为常量。
optimize_for_inference	执行一系列优化传递，以优化模型用于推理。
enable_onednn_fusion	根据参数 enabled 启用或禁用 onednn JIT fusion。
onednn_fusion_enabled	返回 onednn JIT fusion 是否已启用。
set_fusion_strategy	设置在 fusion 过程中可以发生的特殊化的类型和数量。
strict_fusion	如果在推理时并非所有节点都已 fusion，或者在训练时符号区分，则会报错。
保存	保存此模块的离线版本，以便在单独的进程中使用。
加载	加载之前使用 torch.jit.save 保存的 ScriptModule 或 ScriptFunction。
ignore	此装饰器指示编译器忽略函数或方法，并将其保留为 Python 函数。
unused	此装饰器指示编译器忽略函数或方法，并将其替换为引发异常。
interface	装饰以注解不同类型的类或模块。
isinstance	在 TorchScript 中提供容器类型细化。
Attribute	此方法是一个传递函数，返回 value，主要用于向 TorchScript 编译器指示左侧表达式是一个类型为 type 的类实例属性。
annotate	用于在 TorchScript 编译器中指定 the_value 的类型。