torch.fx.passes.pass_manager.log_hook

torch.fx.passes.pass_manager.log_hook(fn, level=20) [исходный код]

Записывает в журнал результат вызываемого объекта.

Это полезно для записи в журнал результатов проходов. Обратите внимание: inplace_wrapper заменяет результат прохода изменённым объектом. Чтобы записать в журнал исходный результат, примените эту обёртку перед inplace_wrapper.

Пример:

def my_pass(d: Dict) -> bool:
    changed = False
    if "foo" in d:
        d["foo"] = "bar"
        changed = True
    return changed


pm = PassManager(passes=[inplace_wrapper(log_hook(my_pass))])

Параметры:

• fn (Callable[Type1, Type2]) –
• level (int) – уровень ведения журнала (например, logging.INFO)

Возвращает:

wrapped_fn (Callable[Type1, Type2])

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

Callable[[~_P], _R]