快捷方式

torch.fx.experimental.proxy_tensor.maybe_enable_thunkify

torch.fx.experimental.proxy_tensor.maybe_enable_thunkify()[源代码][源代码]

在此上下文管理器中,如果您正在进行 make_fx 追踪,我们将 thunkify 所有 SymNode 计算,并避免将其追踪到图中,除非实际需要。您应该尽可能避免使用此功能,因为 SymNode 追踪的延迟评估可能会导致长 thunk 链,如果您评估它们,将会导致堆栈溢出。但是,目前有时这是必要的,因为 PT2 的某些错误部分会因 SymNode 计算追踪不足而导致 “s0 is not tracked with proxy” 错误。

返回类型

Generator[None, None, None]

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源