Constantes
Cadena | DEFAULT_NAME |
Métodos públicos
vacío estático | |
inicio estático |
Métodos heredados
Constantes
Cadena final estática pública DEFAULT_NAME
Métodos públicos
Agregar vacío estático público (alcance de alcance , inicializador de operación )
creación de inicio estática pública (alcance de alcance )
Método de fábrica para crear una operación que ejecuta todos los inicializadores de un gráfico.
Todos los inicializadores agregados a un gráfico mediante tf.initAdd
se agrupan como una única unidad de cálculo en el gráfico. Luego, esta operación debe agregarse a cualquier gráfico usando una o más variables
y ejecutarse una vez antes de ejecutar el gráfico para que los estados de las variables se inicialicen correctamente.
Cuando el gráfico lo crea el mismo proceso que ejecuta la sesión, los inicializadores se pueden invocar ejecutando este único punto final. Por ejemplo:
try (Graph g = new Graph()) {
Variable<TInt32> x = tf.variable(tf.constant(10)); // initAdd is called implicitly
Variable<TInt32> y = tf.variable(tf.constant(20)); // idem
Add<TInt32> z = tf.math.add(x, y);
try (Session s = new Session(g)) {
s.run(tf.init()); // initialize all variables
try (TInt32 t = (TInt32)s.runner().fetch(z).run().get(0)) {
assertEquals(30, t.data().getInt());
}
}
}
Cuando el gráfico se crea mediante un proceso independiente, los inicializadores se pueden invocar ejecutando la operación init por su nombre, que por defecto es DEFAULT_NAME
. Por ejemplo:
// Building the model
try (Graph g = new Graph()) {
Variable<TInt32> x = tf.variable(tf.constant(10)); // initAdd is called implicitly
Variable<TInt32> y = tf.variable(tf.constant(20)); // idem
Add<TInt32> z = tf.withName("z").math.add(x, y);
tf.init(); // add variables initializers to the graph, as Init.DEFAULT_NAME
// ...exporting graph as a saved model...
...
// Running the model
try (SavedModelBundle model = SavedModelBundle.load("/path/to/model", "train")) {
model.session().run(Init.DEFAULT_NAME);
try (TInt32 t = (TInt32)s.runner().fetch("z").run().get(0)) {
assertEquals(30, t.data().getInt());
}
}
}
Parámetros
alcance | alcance actual |
---|
Devoluciones
- una operación que agrupa todos los inicializadores agregados al gráfico
Lanza
Excepción de argumento ilegal | si el entorno de ejecución dentro del alcance no es un gráfico |
---|