EqualityConstraint¶
- class torch.fx.experimental.symbolic_shapes.EqualityConstraint(warn_only, source_pairs, derived_ｅqualities, phantom_symbols, relaxed_sources)[source][source]¶
表示和決定輸入源之間各種型別的相等約束。
“源對”（source pair）是指一對被指定為相等的動態維度的輸入源。我們將 source_pairs 表示在一個並查集（union-find forest）中，以便高效地檢查兩個此類源是否傳遞相等。
“派生相等式”（derived equality）將一個輸入源關聯到根上的一個表示式。該根可以是另一個輸入源（對應於某個動態維度），也可以是不直接表示任何動態維度的虛符號（phantom symbol）。我們將涉及輸入源的 derived_equalities 表示在一個傳遞閉合的對映中，以便高效地檢查一個輸入源是否與另一個輸入源上的給定表示式傳遞相等。（注意：相比之下，很容易判斷一個輸入源是否與虛符號上的給定表示式傳遞相等；此類表示式已經處於規範形式（canonical form），因此問題簡化為符號表達式相等性。）
