@torch.jit.script打包

@torch.jit.script是PyTorch中的一个装饰器，它可以将Python函数编译成Torch脚本，从而提高代码的执行效率，并且可以在不需要Python解释器的情况下执行代码。使用@torch.jit.script装饰器可以将函数包装为Torch Script，然后使用torch.jit.save将其保存到文件中，以便稍后加载和执行。例如：

import torch

@torch.jit.script
def add(x, y):
    return x + y

torch.jit.save(add, "add.pt")

这将add函数编译为Torch脚本，并将其保存到名为“add.pt”的文件中。稍后，您可以使用torch.jit.load加载此脚本并执行它。