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