快捷方式

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。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源