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

返回一個字典，包含符號的名稱及其特定的值

傳回類型

Dict[str, 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()

求解約束方程式系統以找到簡化的約束