捷徑

torch.compiler API 參考

如需 torch.compiler 的快速概覽,請參閱 torch.compiler

compile

如需此函數參數的詳細資訊,請參閱 torch.compile()

reset

此函數會清除所有編譯快取,並將系統還原為初始狀態。

allow_in_graph

告訴編譯器前端 (Dynamo) 跳過函數的符號自省,而是在遇到函數時直接將其寫入圖形。

assume_constant_result

此函數用於將函數 fn 標記為具有常數結果。

list_backends

傳回可以傳遞至 torch.compile(..., backend="name") 的有效字串。

disable

此函數同時提供了一個裝飾器和一個上下文管理器,用於停用函數上的編譯。它還提供了遞迴停用已呼叫函數的選項。

cudagraph_mark_step_begin

表示新的推論或訓練迭代即將開始。

is_compiling

表示是否將圖形執行/追蹤為 torch.compile() 或 torch.export() 的一部分。

is_dynamo_compiling

表示是否透過 TorchDynamo 追蹤圖形。

文件

存取 PyTorch 的完整開發人員文件

檢視文件

教學課程

取得針對初學者和進階開發人員的深入教學課程

檢視教學課程

資源

尋找開發資源並獲得問題的解答

檢視資源