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