快捷方式

EqualityConstraint

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

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

“源对”(source pair)是指一对被指定为相等的动态维度的输入源。我们将 source_pairs 表示在一个并查集(union-find forest)中,以便高效地检查两个此类源是否传递相等。

“派生相等式”(derived equality)将一个输入源关联到根上的一个表达式。该根可以是另一个输入源(对应于某个动态维度),也可以是不直接表示任何动态维度的虚符号(phantom symbol)。我们将涉及输入源的 derived_equalities 表示在一个传递闭合的映射中,以便高效地检查一个输入源是否与另一个输入源上的给定表达式传递相等。(注意:相比之下,很容易判断一个输入源是否与虚符号上的给定表达式传递相等;此类表达式已经处于规范形式(canonical form),因此问题简化为符号表达式相等性。)

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

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

查看资源