torch.fx.experimental.proxy_tensor.maybe_disable_thunkify¶
- torch.fx.experimental.proxy_tensor.maybe_disable_thunkify()[source][source]¶
在上下文管理器中,禁用 thunkification。有关更多详细信息,请参阅
maybe_enable_thunkify()
。这对于您有一个想要启用 thunkification 的包装函数,但在内部某些部分(例如,原始用户函数)中知道不需要 thunkification 时非常有用。- 返回类型
Generator[None, None, None]