评价此页

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, …]]]