DimConstraints

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

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

add(expr)

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

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

Return type

bool

add_equality(source, expr)

添加一个等式约束

forced_specializations()

返回一个从符号名称到其特化值的字典

prettify_results(original_signature, constraint_violation_error=None, forced_specializations=None)

格式化约束违规错误的提示信息

remove_redundant_dynamic_results()

移除形式为 2 <= dynamic_dim(…) 的约束，因为 2 是默认的下限。

rewrite_with_congruences(s, expr)

消除形式为 b // d 和 b % d 的表达式，同时添加形式为 b % d == k 的同余式。 这留下了我们的不等式求解器可以处理的合理运算符（特别是形式为 b / d 的运算符）。 我们分别解决添加的同余式（使用我们的同余求解器，见下文）。

solve(disable_congruences=True, disable_equivalences=True)

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