DimConstraints

class torch.fx.experimental.symbolic_shapes.DimConstraints(symbol_to_source, var_to_val, marked_dynamic, source_name_to_debug_name)

用于符号化维度上约束系统的自定义求解器。解决方案是“静态”值或简化的“动态”约束。

add(expr)

将表达式添加到约束集合中。

返回表达式是否是平凡约束（即，显而易见的重言式）。

返回类型

布尔值

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

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