torch.fx.experimental.proxy_tensor.maybe_disable_thunkify#
- torch.fx.experimental.proxy_tensor.maybe_disable_thunkify()[source]#
在上下文中禁用 thunkification。更多详情请参阅
maybe_enable_thunkify()
。这对于具有您想在其上启用 thunkification 的包装函数非常有用,但在内部的某个片段(例如,原始用户函数)中,您希望禁用 thunkification,因为您知道那里不需要它。- 返回类型
Generator[None, None, None]