評價此頁

torch.fx.experimental.symbolic_shapes.resolve_unbacked_bindings#

torch.fx.experimental.symbolic_shapes.resolve_unbacked_bindings(shape_env, bindings)[原始碼]#

當我們在進行假張量傳播時,我們經常會分配新的未繫結 symint。然後我們執行代理張量模式,該模式會將這些新的 symint 填充到 node.meta["unbacked_bindings"] 中。為了確保一致性,我們使用 PropagateUnbackedSymInts 來將未繫結的繫結重新命名為舊的繫結。但是,所有的 node meta 仍然在使用重新命名之前的舊繫結。此函式有助於事後應用在 PropogateUnbackedSymInts 傳遞中發現的任何重新命名。

返回型別

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