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.