DimConstraints

class torch.fx.experimental.symbolic_shapes.DimConstraints(symbol_to_source, var_to_val, marked_dynamic, source_name_to_debug_name) [исходный код]

Пользовательский решатель для системы ограничений на символические размерности. Решения представляют собой «статические» значения или упрощённые «динамические» ограничения.

add(expr) [исходный код]

Добавляет выражение в набор ограничений.

Возвращает значение, указывающее, является ли выражение тривиальным ограничением (то есть очевидной тавтологией).

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

bool

add_equality(source, expr) [исходный код]

Добавляет ограничение-равенство

forced_specializations() [исходный код]

Возвращает словарь, сопоставляющий имена символов их специализированным значениям

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

dict[str, Expr]

prettify_results(original_signature, dynamic_shapes, constraint_violation_error, forced_specializations) [исходный код]

Форматирует сообщение об ошибках нарушения ограничений

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

str

rewrite_with_congruences(s, expr) [исходный код]

Устраняет выражения вида b // d и b % d, добавляя сравнения по модулю вида b % d == k. В результате остаются рациональные операторы (в частности, вида b / d), с которыми может работать наш решатель неравенств. Добавленные сравнения решаются отдельно (с помощью нашего решателя сравнений по модулю, см. ниже).

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

_SympyT

solve() [исходный код]

Решает систему уравнений-ограничений, чтобы найти упрощённые ограничения