tf.compat.v1.wrap_function

Оборачивает функцию TF 1.x fn в функцию графа.

tf.compat.v1.wrap_function(
    fn, signature, name=None
)

Python-функция fn будет вызвана один раз с символичными аргументами, указанными в signature, отслежена и преобразована в функцию графа. Любые переменные, созданные функцией fn, будут принадлежать объекту, возвращённому функцией wrap_function. Полученная функция графа может быть вызвана с тензорами, соответствующими её сигнатуре.

И tf.compat.v1.wrap_function, и tf.function создают вызываемую функцию TensorFlow графа. Но в то время как tf.function выполняет все операции со состоянием (например, tf.print) и последовательность операций, обеспечивающих те же семантику, что и устремлённое выполнение, wrap_function ближе к поведению session.run в TensorFlow 1.x. Она не будет выполнять какие-либо операции, если они не требуются для вычисления выходных значений функции, либо через зависимость от данных, либо через зависимость управления. Она также не будет упорядочивать операции.

В отличие от tf.function, wrap_function будет отслеживать Python-функцию только один раз. Как и в случае с плейсхолдерами в TF 1.x, формы и типы данных должны быть предоставлены в аргументе wrap_function's signature.

Поскольку отслеживание происходит только один раз, переменные и состояние могут быть созданы внутри функции и принадлежать объекту-обёртке функции.