快捷方式

torch.compiler API 參考

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

compile

請參閱 torch.compile() 以取得有關此函數參數的詳細資訊。

reset

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

allow_in_graph

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

substitute_in_graph

為函數註冊一個 polyfill 處理程序,通常是來自 C 擴展的 C 函數,以便在圖表中內聯原始函數時取代原始函數。

assume_constant_result

此函式用於將函式 fn 標記為具有恆定結果。

list_backends

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

disable

此函式提供一個裝飾器來停用函式的編譯。 它還提供遞迴停用所呼叫函式的選項。

set_stance

設定編譯器的目前姿態 (stance)。

cudagraph_mark_step_begin

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

is_compiling

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

is_dynamo_compiling

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

文件

存取 PyTorch 的全面開發者文件

檢視文件

教學

取得適合初學者和進階開發者的深入教學

檢視教學

資源

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

檢視資源