torch.fx.experimental.proxy_tensor.maybe_disable_thunkify¶
- torch.fx.experimental.proxy_tensor.maybe_disable_thunkify()[源代码]¶
在上下文中禁用 thunkification。有关更多详细信息,请参阅
maybe_enable_thunkify()
。如果您有一个包装函数,您希望在其上启用 thunkification,但在内部的某些部分(例如,原始用户函数),您希望禁用 thunkification,因为您知道在那里不需要它,这将很有帮助。- 返回类型
生成器[None, None, None]