快捷方式

EqualityConstraint

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

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

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

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源