functorch.compile.ts_compile

functorch.compile.ts_compile(fx_g, inps)

使用 Torchscript 編譯器編譯 fx_g。

警告

此 API 仍處於實驗階段，可能會有所變更。

參數

fx_g (fx.GraphModule) – 要編譯的輸入 Fx 圖形模組。

回傳值

Torch 腳本模型。