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

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