评价此页

torch.fx.experimental.symbolic_shapes.find_symbol_binding_fx_nodes#

torch.fx.experimental.symbolic_shapes.find_symbol_binding_fx_nodes(graph)[source]#

查找绑定 sympy Symbol 的 FX 图中的所有节点。

此函数扫描给定 FX 图中的所有节点,并识别绑定 sympy Symbol 的节点(通常是具有 SymInt 值的占位符节点)。当多个节点绑定同一个 symbol 时,只保留第一次出现。

参数:

graph (Graph) – 要搜索 symbol 绑定节点的 FX 图

返回:

一个字典,将 sympy Symbol 映射到它们绑定的 FX 节点

返回类型:

dict[Symbol, Node]