torch.fx.experimental.symbolic_shapes.rebind_unbacked¶
- torch.fx.experimental.symbolic_shapes.rebind_unbacked(shape_env, n, result)[源][源]¶
假設我們正在重新追蹤先前已存在且執行過 fake tensor 傳播(因此包含 unbacked SymInts)的 FX 圖。當我們重新追蹤時,我們會重新傳播 fake tensor,這將產生新的 unbacked SymInts。此時,我們需要通知形狀環境舊的 unbacked SymInts 與新的 unbacked SymInts 之間的等價關係。請向我們傳遞舊的 torch.fx.Node(包含舊的繫結資訊)和新的結果(我們可以從中提取新的 unbacked SymInts)。