Torchscript提供torch.jit.trace和torch.jit.script将pytorch代码从热切模式转换为脚本模型。从文档中，我可以理解torch.jit.trace无法处理python中存在的控制流和其他数据结构。因此，开发了torch.jit.script来解决torch.jit.trace中的问题。
但是看起来torch.jit.script适用于所有情况，那么为什么我们需要torch.jit.trace？
请帮助我了解这两种方法之间的区别
答案 0 :(得分：0)
如果else适用于您的代码，那么这就是您所需要的。 torch.jit.script使用的编译器不支持使用动态行为（例如多态）的代码，因此在这种情况下，您需要使用torch.jit.script。