EqualityConstraint#
- class torch.fx.experimental.symbolic_shapes.EqualityConstraint(warn_only, source_pairs, derived_equalities, phantom_symbols, relaxed_sources)[原始碼]#
表示並決定輸入源之間的各種等價約束。
“源對”是為指定相等的動態尺寸的輸入源對。我們將source_pairs表示在並查集中,以便我們可以有效地檢查兩個這樣的源是否具有傳遞性等價性。
“派生等式”將一個輸入源與一個根上的表示式關聯起來。根可以是另一個輸入源(對應於某個動態尺寸),也可以是一個不直接表示任何動態尺寸的幻影符號。我們將涉及輸入源的derived_equalities表示在一個傳遞閉包對映中,以便我們可以有效地檢查一個輸入源是否與另一個輸入源上的給定表示式具有傳遞性等價性。(注意:相比之下,可以很容易地確定一個輸入源是否與幻影符號上的給定表示式具有傳遞性等價性;這樣的表示式已經處於規範形式,因此問題就簡化為符號表達式等價性。)