Se usó la API de Cloud Translation para traducir esta página.
Switch to English

tf.compat.v1.wrap_function

Ver código fuente en GitHub

Envuelve la función fn TF 1.x en una función de gráfico.

Se utiliza en los cuadernos

Se utiliza en la guía

La función pitón fn será llamada una vez con argumentos simbólicos especificado en la signature , rastreado, y se convirtió en una función de gráfico. Cualquiera de las variables creadas por fn serán propiedad de objeto devuelto por wrap_function . La función gráfica resultante se puede llamar con tensores que coincide con la firma.

 def f(x, do_add):
  v = tf.Variable(5.0)
  if do_add:
    op = v.assign_add(x)
  else:
    op = v.assign_sub(x)
  with tf.control_dependencies([op]):
    return v.read_value()

f_add = tf.compat.v1.wrap_function(f, [tf.TensorSpec((), tf.float32), True])

assert float(f_add(1.0)) == 6.0
assert float(f_add(1.0)) == 7.0

# Can call tf.compat.v1.wrap_function again to get a new trace, a new set
# of variables, and possibly different non-template arguments.
f_sub= tf.compat.v1.wrap_function(f, [tf.TensorSpec((), tf.float32), False])

assert float(f_sub(1.0)) == 4.0
assert float(f_sub(1.0)) == 3.0
 

Tanto tf.compat.v1.wrap_function y tf.function crear un gráfico TensorFlow exigible. Pero mientras tf.function ejecuta todas las operaciones con estado (por ejemplo tf.print ) y secuencias de operaciones para proporcionar la misma semántica que la ejecución ansiosos, wrap_function está más cerca del comportamiento de session.run en TensorFlow 1.x. No se ejecutará ninguna operación a menos que se requieren para calcular las salidas de la función, ya sea a través de una dependencia de datos o una dependencia de control. Ni va a secuenciar las operaciones.

A diferencia de tf.function , wrap_function sólo se trazará la función Python vez. Al igual que con marcadores de posición en 1.x TF, formas y dtypes deben ser proporcionados a wrap_function 's signature argumento.

Ya que sólo se traza una vez, variables y estado pueden ser creados dentro de la función y propiedad del objeto función de contenedor.

fn función pitón para ser envuelto
signature los argumentos de marcador de posición y Python que se pasan a la función envuelto
name Opcional. El nombre de la función.

la función de gráfico envuelto.