torch.fx.experimental.symbolic_shapes.find_symbol_binding_fx_nodes

torch.fx.experimental.symbolic_shapes.find_symbol_binding_fx_nodes(graph) [исходный код]

Находит все узлы графа FX, связывающие символы sympy.

Эта функция просматривает все узлы указанного графа FX и определяет узлы, связывающие символы sympy (обычно узлы-заполнители со значениями SymInt). Если несколько узлов связывают один и тот же символ, сохраняется только первое вхождение.

Параметры:

graph (Graph) – Граф FX, в котором нужно найти узлы, связывающие символы

Возвращает:

Словарь, сопоставляющий символы sympy с соответствующими связывающими их узлами FX

Тип возвращаемого значения:

dict[Symbol, Node]