DimConstraints

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

符号维度约束系统的自定义求解器。解为“静态”值或简化的“动态”约束。

添加(expr)[来源][来源] ¶

向约束集合中添加一个表达式。

返回表达式是否是平凡约束（即明显的恒真命题）。

返回类型:

布尔型

添加等式（source，expr）[source][source] ¶

添加一个等式约束

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 == k 的同余关系的同时，消除形式为 b // d 和 b % d 的表达式。这留下了我们的不等式求解器可以处理的理性运算符（特别是形式为 b / d 的运算符）。我们单独解决添加的同余关系（使用我们的同余求解器，见下文）。

返回类型:

_SympyT

solve()

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