TorchScript

创建于：2018年9月7日 | 最后更新于：2025年7月16日

警告

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

创建 TorchScript 代码

script	脚本化函数。
trace	跟踪（Trace）一个函数并返回一个可执行的或 ScriptFunction，它将通过即时编译进行优化。
script_if_tracing	在跟踪过程中首次调用时编译（script）fn。
trace_module	跟踪（Trace）一个模块并返回一个可执行的 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 融合。
onednn_fusion_enabled	返回 onednn JIT 融合是否已启用。
set_fusion_strategy	设置融合过程中可能发生的特化类型和数量。
strict_fusion	如果在推理时并非所有节点都已融合，或者在训练时进行了符号微分，则会引发错误。
保存	保存此模块的离线版本，以便在单独的进程中使用。
加载	加载之前使用 torch.jit.save 保存的 ScriptModule 或 ScriptFunction。
ignore	此装饰器指示编译器忽略函数或方法，并将其保留为 Python 函数。
unused	此装饰器指示编译器忽略函数或方法，并将其替换为引发异常。
interface	使用装饰器来注解不同类型的类或模块。
isinstance	在 TorchScript 中提供容器类型细化。
Attribute	此方法是一个传递函数，返回 value，主要用于向 TorchScript 编译器指示左侧表达式是一个具有 type 类型的类实例属性。
annotate	用于在 TorchScript 编译器中给定 the_value 的类型。