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