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

求解約束方程組以找到簡化的約束