评价此页

torch.fx.experimental.symbolic_shapes.rebind_unbacked#

torch.fx.experimental.symbolic_shapes.rebind_unbacked(shape_env, n, result)[source]#

假设我们正在重新追踪一个先前具有虚张量传播(因此是非绑定 SymInts)的 FX 图。当我们重新追踪时,我们会重新传播虚张量,这会产生新的非绑定 SymInts。发生这种情况时,我们需要告知形状环境旧的非绑定 SymInts 和新的非绑定 SymInts 之间的等价性。将旧的 torch.fx.Node(具有旧的绑定信息)和新的结果(我们可以从中提取新的非绑定 SymInts)传递给我们。