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, sympy.core.expr.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()

求解约束方程系统以找到简化的约束