torch.fx.experimental.symbolic_shapes.rebind_unbacked¶
- torch.fx.experimental.symbolic_shapes.rebind_unbacked(shape_env, n, result)[源][源]¶
假设我们正在重新追踪先前已存在且执行过 fake tensor 传播(因此包含 unbacked SymInts)的 FX 图。当我们重新追踪时,我们会重新传播 fake tensor,这将产生新的 unbacked SymInts。此时,我们需要通知形状环境旧的 unbacked SymInts 与新的 unbacked SymInts 之间的等价关系。请向我们传递旧的 torch.fx.Node(包含旧的绑定信息)和新的结果(我们可以从中提取新的 unbacked SymInts)。