快捷方式

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

設定編譯器的當前立場。

cudagraph_mark_step_begin

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

is_compiling

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

is_dynamo_compiling

指示圖是否透過 TorchDynamo 進行跟蹤。

is_exporting

指示是否正在匯出。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲取問題解答

檢視資源