EqualityConstraint#
- class torch.fx.experimental.symbolic_shapes.EqualityConstraint(warn_only, source_pairs, derived_equalities, phantom_symbols, relaxed_sources)[source]#
表示并决定输入源之间的各种相等约束。
“源对”是动态维度的输入源对,它们被指定为相等。我们将source_pairs表示在一个并查集森林中,以便能够有效地检查两个这样的源是否是传递相等的。
“派生相等”将一个输入源与一个根上的表达式关联起来。根可以是另一个输入源,对应于某个动态维度,或者是一个不直接表示任何动态维度的幻影符号。我们将涉及输入源的derived_equalities表示在一个传递闭包的映射中,以便能够有效地检查一个输入源是否传递地等于另一个输入源上的表达式。 (注意:相比之下,很容易确定一个输入源是否传递地等于一个幻影符号上的表达式;这样的表达式已经是规范形式,因此问题就简化为符号表达式相等性。)