评价此页

EqualityConstraint#

class torch.fx.experimental.symbolic_shapes.EqualityConstraint(warn_only, source_pairs, derived_equalities, phantom_symbols, relaxed_sources)[源代码]#

表示并决定输入源之间的各种等价约束。

“源对”是为指定相等的动态尺寸的输入源对。我们将source_pairs表示在并查集中,以便我们可以有效地检查两个这样的源是否具有传递性等价性。

“派生等式”将一个输入源与一个根上的表达式关联起来。根可以是另一个输入源(对应于某个动态尺寸),也可以是一个不直接表示任何动态尺寸的幻影符号。我们将涉及输入源的derived_equalities表示在一个传递闭包映射中,以便我们可以有效地检查一个输入源是否与另一个输入源上的给定表达式具有传递性等价性。(注意:相比之下,可以很容易地确定一个输入源是否与幻影符号上的给定表达式具有传递性等价性;这样的表达式已经处于规范形式,因此问题就简化为符号表达式等价性。)