快捷方式

torch.fx.experimental.symbolic_shapes.rebind_unbacked

torch.fx.experimental.symbolic_shapes.rebind_unbacked(shape_env, n, result)[源][源]

假设我们正在重新追踪先前已存在且执行过 fake tensor 传播(因此包含 unbacked SymInts)的 FX 图。当我们重新追踪时,我们会重新传播 fake tensor,这将产生新的 unbacked SymInts。此时,我们需要通知形状环境旧的 unbacked SymInts 与新的 unbacked SymInts 之间的等价关系。请向我们传递旧的 torch.fx.Node(包含旧的绑定信息)和新的结果(我们可以从中提取新的 unbacked SymInts)。

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源