torch.jit.script（）和 torch.jit.trace()区别

---

前言

torch.jit.script() 和 torch.jit.trace() 是 PyTorch 中用于将模型转换为 TorchScript 表示的两种不同方法。

---

1、torch.jit.script():

• 完整转换：torch.jit.script() 将整个模型转换为 TorchScript
  形式，包括模型的定义、前向传播函数以及任何与模型相关的操作、控制流和参数。
• 动态控制流支持：torch.jit.script() 可以处理动态控制流语句和复杂的操作，因为它将整个模型转换为 TorchScript形式。
• 需要模型定义：torch.jit.script() 需要模型的源代码或模型类的定义，因此需要具有完整的模型定义。
• 适用于复杂模型：适用于包含复杂逻辑、条件语句和循环的模型，并且可以处理更多类型的操作。

2、torch.jit.trace():

• 前向传播跟踪：torch.jit.trace() 仅跟踪模型的前向传播函数，而不是整个模型。
• 静态图形表示：跟踪后的模型是基于模型的前向传播过程生成的计算图，它是模型的静态图形表示。
• 不需要完整模型定义：torch.jit.trace()
  不需要完整的模型定义，只需要提供模型实例和示例输入，它会通过示例输入的前向传播过程进行跟踪。
• 适用于简单模型：适用于简单的模型，不涉及复杂的条件语句或循环，并且可以处理的操作类型较为有限。