評價此頁

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)[原始碼]#

在運行了 fake tensor 傳播併產生 example_value 結果後,遍歷 example_value 來查詢新繫結的未備份符號,並記錄它們的路徑以供後續使用。如果我們分配了一個未備份的 SymInt 但在 example_value 中找不到它,則會出錯。(注意:這意味著如果您有一個多輸出函式,您必須在張量輸出的元組上呼叫此函式,您不能等待!)

peek 引數允許您檢視繫結內容,而無需更改受影響的列表。這主要用於確保在 propagate_real_tensors 開啟時,unbacked_var_to_val 能夠及時填充。

返回型別

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