1、TorchScript 是 Python 的静态类型子集,可以直接编写(使用@torch.jit.script装饰器)或者通过跟踪(tracing)从python代码自动生成。当使用跟踪(tracing)时,代码会通过仅仅记录张量上的真实OP操作,来自动转换为 Python 的子集;并简单地执行和丢弃其他周围的 Python 代码。
注:python装饰器学习可以参考:链接1、链接2。
当直接使用 @torch.jit.script装饰
@torch.jit.script装饰
京公网安备 11010502049817号