torch.fx.experimental.proxy_tensor.make_fx

torch.fx.experimental.proxy_tensor.make_fx(f, decomposition_table=None, tracing_mode='real', _allow_non_fake_inputs=False, *, pre_dispatch=False, record_module_stack=False, _allow_fake_constant=False, _error_on_data_dependent_ops=True, record_stack_traces=False, proxy_module_inputs=False, _disable_torch_fn_metadata_mode=False, dynamic_shapes=None) [исходный код]

Для заданной функции f возвращает новую функцию, которая при вызове с допустимыми аргументами для f возвращает FX GraphModule, представляющий набор операций, выполненных в ходе выполнения.

Если record_stack_traces имеет значение True, трассировка стека будет сохранена в node.meta[“stack_trace”]

По умолчанию трассировки post-dispatch без явной перезаписи через decomposition_table заменяют detach на alias. Повторное дифференцирование возвращённого графа не сохраняет эти вызовы detach как границы отсечения градиента autograd; передайте явную decomposition_table или используйте pre_dispatch=True, если нужны точные узлы detach.

tracing_mode:

• "real": без fakification, трассировка выполняется с реальными тензорами.
• "fake": тензоры становятся FakeTensors. По умолчанию все размерности являются статическими (конкретными); если задан dynamic_shapes, спецификация определяет, какие размерности становятся неподкреплёнными символическими размерами — все размерности, не объявленные в спецификации, остаются статическими.
• "symbolic": тензоры становятся FakeTensors, и каждая размерность становится подкреплённым символическим размером. ПРИМЕЧАНИЕ: использовать подкреплённые символы таким образом, как правило, не рекомендуется и небезопасно — подкреплённые символы предполагают определённые условия и могут незаметно ограничиваться защитными проверками. Сохранено для обратной совместимости. Предпочтительно использовать "fake" + dynamic_shapes, чтобы явно объявлять динамичность с помощью неподкреплённых символов.

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

Callable[[…], GraphModule]