tf.compat.v1.flags.tf_decorator.rewrap

Вставляет новую цель в функцию, созданную с помощью make_decorator.

tf.compat.v1.flags.tf_decorator.rewrap(
    decorator_func, previous_target, new_target
)

Эта функция позволяет заменить функцию, обернутую с помощью decorator_func, предполагая, что декоратор, который оборачивает функцию, написан как описано ниже.

Функция-декоратор должна использовать <decorator name>.__wrapped__ вместо оборачиваемой функции, которая обычно используется:

Пример:

Вместо этого:

def simple_parametrized_wrapper(*args, *kwds): return wrapped_fn(args, **kwds)

tf_decorator.make_decorator(simple_parametrized_wrapper, wrapped_fn)

Напишите это:

def simple_parametrized_wrapper(*args, *kwds): return simple_parametrizedwrapper.wrapped_(args, **kwds)

tf_decorator.make_decorator(simple_parametrized_wrapper, wrapped_fn)

Обратите внимание, что этот процесс изменяет decorator_func.