EqualityConstraint

class torch.fx.experimental.symbolic_shapes.EqualityConstraint(warn_only, source_pairs, derived_equalities, phantom_symbols, relaxed_sources)

表示和决定输入源之间各种类型的相等约束。

“源对”是指指定为相等的动态维度的输入源对。我们在并查森林中表示 source_pairs，以便我们可以有效地检查两个这样的源是否传递相等。

“派生相等”将输入源与根上的表达式相关联。根可以是另一个输入源（对应于某些动态维度），也可以是不直接表示任何动态维度的幻象符号。我们在传递闭包映射中表示涉及输入源的 derived_equalities，以便我们可以有效地检查输入源是否传递相等于给定表达式（在另一个输入源上）。（注意：相比之下，很容易确定输入源是否传递相等于给定表达式（在幻象符号上）；此类表达式已经采用规范形式，因此问题简化为符号表达式相等性。）