torch.fx.experimental.symbolic_shapes.compute_unbacked_bindings¶
- torch.fx.experimental.symbolic_shapes.compute_unbacked_bindings(shape_env, example_value, old_example_value=None, peek=False)[源代码][源代码]¶
在运行伪张量传播并生成 example_value 结果后，遍历 example_value 以查找新绑定的未备份符号，并记录它们的路径以供后续使用。如果我们分配了一个未备份的 SymInt 但在 example_value 中找不到它，则会报错。（注意：这意味着如果你有一个多输出函数，你必须对张量输出元组调用此函数，不能等待！）
peek 参数允许你检查绑定是什么，而无需更改受影响的列表。这主要用于确保当 propagate_real_tensors 开启时，unbacked_var_to_val 能够及时填充。
