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)[原始碼][原始碼]¶
在執行過虛假張量傳播並產生 example_value 結果之後,遍歷 example_value 尋找新綁定的無後盾符號,並記錄它們的路徑以供稍後使用。如果我們已經分配了一個無後盾的 SymInt,但它無法在 example_value 中找到,則會發生錯誤。(注意:這表示如果您有一個多輸出函數,您必須在張量輸出的元組上調用此函數,您不能等待!)
peek 參數可讓您檢查綁定是什麼,而無需更改受影響的列表。這主要用於確保當 propagate_real_tensors 開啟時,unbacked_var_to_val 會立即填入。