functorch.compile.ts_compile

functorch.compile.ts_compile(fx_g, inps)

使用 Torchscript 编译器编译 fx_g。

警告

此 API 处于实验阶段，可能会发生更改。

参数

fx_g (fx.GraphModule) – 要编译的输入 Fx 图模块。

返回值

Torch 脚本模型。