torch.fx.experimental.proxy_tensor.maybe_disable_thunkify
-
torch.fx.experimental.proxy_tensor.maybe_disable_thunkify()[source][source]
在上下文中,禁用 thunkification。参见 maybe_enable_thunkify()
获取更多详情。如果您有一个想要启用 thunkification 的包装函数,但在内部某些部分(比如原始用户函数)中您知道不需要 thunkification,这将很有帮助。
- 返回类型:
生成器[None, None, None]