torch.fx.experimental.proxy_tensor.maybe_enable_thunkify

torch.fx.experimental.proxy_tensor.maybe_enable_thunkify() [исходный код]

В рамках этого менеджера контекста при трассировке с помощью make_fx мы преобразуем все вычисления SymNode в thunks и не будем трассировать их в граф, если это действительно не потребуется. По возможности старайтесь не использовать этот механизм: отложенная трассировка SymNode может привести к длинным цепочкам thunks, которые вызовут переполнение стека при вычислении. Однако иногда это необходимо, поскольку из-за ошибок в некоторых частях PT2 возникает ошибка «s0 is not tracked with proxy» в результате недостаточной трассировки вычислений SymNode.

Тип возвращаемого значения:

Generator[None, None, None]