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

返回一个字典，其中包含符号名称及其专门值。

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

格式化约束违反错误的消息

rewrite_with_congruences(s, expr)

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

solve()

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