TorchScript

建立日期: 2018 年 9 月 7 日 | 最後更新日期: 2025 年 7 月 16 日

警告

TorchScript 已棄用，請使用 torch.export 代替。

建立 TorchScript 程式碼

script	指令碼化函式。
trace	跟蹤函式並返回一個可執行函式或 ScriptFunction，它將透過即時編譯進行最佳化。
script_if_tracing	在跟蹤期間首次呼叫 fn 時進行編譯。
trace_module	跟蹤模組並返回一個可執行 ScriptModule，它將透過即時編譯進行最佳化。
fork	建立一個執行 func 的非同步任務，並返回該執行結果的引用。
wait	強制完成 torch.jit.Future[T] 非同步任務，並返回任務的結果。
ScriptModule	C++ torch::jit::Module 的包裝器，包含方法、屬性和引數。
ScriptFunction	功能上等同於 ScriptModule，但它代表單個函式，沒有任何屬性或引數。
freeze	凍結 ScriptModule，將子模組和屬性內聯為常量。
optimize_for_inference	執行一系列最佳化傳遞，以最佳化模型用於推理。
enable_onednn_fusion	根據引數 enabled 啟用或停用 onednn JIT fusion。
onednn_fusion_enabled	返回 onednn JIT fusion 是否已啟用。
set_fusion_strategy	設定在 fusion 過程中可以發生的特殊化的型別和數量。
strict_fusion	如果在推理時並非所有節點都已 fusion，或者在訓練時符號區分，則會報錯。
儲存	儲存此模組的離線版本，以便在單獨的程序中使用。
載入	載入之前使用 torch.jit.save 儲存的 ScriptModule 或 ScriptFunction。
ignore	此裝飾器指示編譯器忽略函式或方法，並將其保留為 Python 函式。
unused	此裝飾器指示編譯器忽略函式或方法，並將其替換為引發異常。
interface	裝飾以註解不同型別的類或模組。
isinstance	在 TorchScript 中提供容器型別細化。
Attribute	此方法是一個傳遞函式，返回 value，主要用於向 TorchScript 編譯器指示左側表示式是一個型別為 type 的類例項屬性。
annotate	用於在 TorchScript 編譯器中指定 the_value 的型別。