torch.fx.experimental.symbolic_shapes.rebind_unbacked
-
torch.fx.experimental.symbolic_shapes.rebind_unbacked(shape_env, n, result)[source][source]
假设我们正在回溯一个之前存在过伪造张量传播(因此没有后盾的 SymInts)的 FX 图。当我们回溯时,我们会重新传播伪造的张量,这会导致新的没有后盾的 SymInts。当这种情况发生时,我们需要告诉形状环境关于新旧没有后盾的 SymInts 等价性的信息。请传递给我们旧的 torch.fx.Node(它包含旧的绑定信息)以及新的结果(我们可以从中提取新的没有后盾的 SymInts)。