快捷键

等价约束 ¶

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

表示和决定输入源之间各种等价约束。

“源对”是一对动态维度的输入源,它们被指定为相等。我们使用并查集森林来表示源对,以便可以高效地检查两个这样的源是否是传递等价的。

“派生等价”将一个输入源与一个根表达式相关联。根可以是另一个输入源,对应某个动态维度,或者一个不直接表示任何动态维度的幻影符号。我们使用传递闭包映射来表示涉及输入源的派生等价,以便可以高效地检查一个输入源是否与另一个输入源上的给定表达式传递等价。(注意:与输入源相关的派生等价,如果涉及幻影符号,则很容易判断是否传递等价;这样的表达式已经处于规范形式,因此问题简化为符号表达式等价。)


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

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

查看资源