评价此页

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)传递给我们。