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()

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