DimConstraints¶

class torch.fx.experimental.symbolic_shapes.DimConstraints(symbol_to_source, var_to_val, marked_dynamic, source_name_to_debug_name)[source][source]¶

符號維度約束系統的自定義求解器。解是“靜態”值或簡化的“動態”約束。

add(expr)[source][source]¶

新增一個表示式到約束集中。

返回該表示式是否為平凡約束（即，明顯的重言式）。

返回型別: bool

add_equality(source, expr)[source][source]¶

新增一個等式約束

forced_specializations()[source][source]¶

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

返回型別: dict[str, sympy.core.expr.Expr]

prettify_results(original_signature, dynamic_shapes, constraint_violation_error, forced_specializations)[source][source]¶

格式化約束違反錯誤的提示資訊

返回型別: str

rewrite_with_congruences(s, expr)[source][source]¶

消除 b // d 和 b % d 形式的表示式，同時新增 b % d == k 形式的同餘關係。這會留下我們的不等式求解器可以處理的有理數運算元（特別是 b / d 形式）。我們單獨求解新增的同餘關係（使用我們的同餘求解器，參見下文）。

返回型別: _SympyT

solve()[source][source]¶

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

DimConstraints¶

文件

教程

資源