评价此页

torch.fx.experimental.symbolic_shapes.resolve_unbacked_bindings#

torch.fx.experimental.symbolic_shapes.resolve_unbacked_bindings(shape_env, bindings)[source]#

在进行fake tensor传播时,我们经常会分配新的未绑定符号整数。然后我们运行代理张量模式,该模式会将node.meta["unbacked_bindings"]填充为这些新的符号整数。为了确保一致性,我们使用PropagateUnbackedSymInts来将未绑定的绑定重命名为旧的绑定。但是所有的节点元数据仍然使用重命名之前的旧绑定。此函数有助于事后应用PropogateUnbackedSymInts传递中发现的任何重命名。

返回类型

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