評價此頁

torch.fx.experimental.symbolic_shapes.rebind_unbacked#

torch.fx.experimental.symbolic_shapes.rebind_unbacked(shape_env, n, result)[原始碼]#

假設我們正在重新追蹤一個已存在的 FX 圖,該圖先前進行了虛假張量傳播(因此存在未繫結 SymInts)。當我們重新追蹤時,我們會重新傳播虛假張量,這會產生新的未繫結 SymInts。發生這種情況時,我們需要告知形狀環境舊的和新的未繫結 SymInts 的等價性。將舊的 torch.fx.Node(具有舊的繫結資訊)和新的結果(我們可以從中提取新的未繫結 SymInts)傳遞給我們。