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 及時填充。