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)[源代码]#
在运行了 fake tensor 传播并生成 example_value 结果后,遍历 example_value 以查找新绑定的未支持符号,并记录它们的路径以供后续使用。如果在分配了一个未支持的 SymInt 后在 example_value 中找不到它,则会报错。(注意:这意味着如果你有一个多输出函数,你必须对其元组输出调用此函数,不能等待!)
peek 参数允许你在不更改受影响列表的情况下查看绑定。这主要用于确保在 propagate_real_tensors 开启时,unbacked_var_to_val 能够及时填充。