快捷方式

torch.fx.experimental.symbolic_shapes.compute_unbacked_bindings

torch.fx.experimental.symbolic_shapes.compute_unbacked_bindings(shape_env, example_value, old_example_value=None, peek=False)[source][source]

在運行了假張量傳播併產生 example_value 結果後,遍歷 example_value,尋找新繫結的未支援符號(unbacked symbols)並記錄其路徑以備後用。如果分配了未支援的 SymInt 但在 example_value 中找不到,則會出錯。(注意:這意味著如果有一個多輸出函式,必須在張量輸出的元組上呼叫此函式,不能等待!)

peek 引數允許您在不改變受影響列表的情況下檢視繫結內容。這主要用於確保在啟用 propagate_real_tensors 時 unbacked_var_to_val 及時填充。

返回型別

Optional[dict[sympy.core.symbol.Symbol, tuple[torch.utils._pytree.KeyEntry, …]]]


© 版權所有 PyTorch 貢獻者。

使用 Sphinx 構建,主題由 Read the Docs 提供。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並解答問題

檢視資源