评价此页

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]