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