快捷方式

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][source]

在运行了假张量传播并产生 example_value 结果后,遍历 example_value,寻找新绑定的未支持符号(unbacked symbols)并记录其路径以备后用。如果分配了未支持的 SymInt 但在 example_value 中找不到,则会出错。(注意:这意味着如果有一个多输出函数,必须在张量输出的元组上调用此函数,不能等待!)

peek 参数允许您在不改变受影响列表的情况下查看绑定内容。这主要用于确保在启用 propagate_real_tensors 时 unbacked_var_to_val 及时填充。

返回类型

Optional[dict[sympy.core.symbol.Symbol, tuple[torch.utils._pytree.KeyEntry, …]]]


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,主题由 Read the Docs 提供。

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

查找开发资源并解答问题

查看资源