Driver para Graph
ejecución.
Una Session
ejemplo encapsula el entorno en el que Operation
s en un Graph
son ejecutados para calcular Tensors
. Por ejemplo:
// Let's say graph is an instance of the Graph class
// for the computation y = 3 * x
try (Session s = new Session(graph)) {
try (Tensor x = Tensor.create(2.0f);
Tensor y = s.runner().feed("x", x).fetch("y").run().get(0)) {
System.out.println(y.floatValue()); // Will print 6.0f
try (Tensor x = Tensor.create(1.1f);
Tensor y = s.runner().feed("x", x).fetch("y").run().get(0)) {
System.out.println(y.floatValue()); // Will print 3.3f
}
}
}
ADVERTENCIA: Una Session
es propietaria de los recursos que deben ser liberados de forma explícita invocando close()
.
Las instancias de una sesión son seguras para subprocesos.
Clases anidadas
clase | Session.Run | Tensores de salida y metadatos obtenidos al ejecutar una sesión. | |
clase | Session.Runner | Ejecutar Operation s y evaluar Tensors . |
Constructores públicos
Sesión ( Graph g, ConfigProto config) Construir una nueva sesión con los asociados Graph opciones y configuración. |
Métodos públicos
vacío | Cerrar () Liberar recursos asociados con la sesión. |
vacío | restaurar (String prefijo) Restaurar el estado actual de las variables del gráfico de esta sesión. |
vacío | |
vacío | ejecutar (String opname) Ejecuta una operación en el gráfico con el nombre de pila. |
vacío | runInit () Ejecute los inicializadores del gráfico. |
Session.Runner | corredor () Cree un corredor para ejecutar operaciones gráficas y evaluar tensores. |
vacío | Guardar (String prefijo) Guarda el estado actual de las variables del gráfico de esta sesión. |
Métodos heredados
Constructores públicos
Sesión pública ( Gráfico g)
Sesión pública ( Gráfico g, ConfigProto config)
Construir una nueva sesión con los asociados Graph
opciones y configuración.
Parámetros
gramo | El Graph la sesión creada operará en. |
---|---|
config | Parámetros de configuración de la sesión especificada como ConfigProto búfer de protocolo. |
Lanza
Argumento de excepción ilegal | si la configuración no es una serialización válida del búfer del protocolo ConfigProto. |
---|
Métodos públicos
void close pública ()
Liberar recursos asociados con la sesión.
Bloques hasta que no haya ejecuciones activas ( run()
llamadas). Una sesión no se puede utilizar después de cerrar devoluciones.
public void restauración (cadena de prefijo)
Restaurar el estado actual de las variables del gráfico de esta sesión.
prefix
es la ruta donde los archivos que contienen las variables de estado en vivo, seguido por el prefijo del nombre. Por ejemplo, si prefix
se establece en MyModel / myvariables / variables, a continuación, los archivos se cargan desde MyModel / myvariables y nombrado variables.data - * - * de-
Tenga en cuenta que este método podría alterar el gráfico subyacente si es la primera vez que una de sus sesiones se guarda, ver ERROR(/Graph#saverDef())
para más detalles.
Parámetros
prefijo | prefijo para restaurar desde |
---|
public void run ( Op op)
Ejecuta una operación en el gráfico.
Este método es equivalente a session.runner().addTarget(op).run()
.
Parámetros
op | la operación a ejecutar. |
---|
public void run (String opname)
Ejecuta una operación en el gráfico con el nombre de pila.
Este método es equivalente a session.runner().addTarget(opName).run()
.
Parámetros
opName | nombre de la operación a ejecutar. |
---|
Lanza
Argumento de excepción ilegal | si no se puede encontrar ninguna operación con ese nombre en el gráfico |
---|
runInit public void ()
Ejecute los inicializadores del gráfico.
Este método es equivalente a session.run(Ops.create(session.graph).init())
.
pública Session.Runner corredor ()
Cree un corredor para ejecutar operaciones gráficas y evaluar tensores.
public void Save (cadena de prefijo)
Guarda el estado actual de las variables del gráfico de esta sesión.
prefix
es una ruta donde se guardarán los archivos que contienen el estado de las variables, seguidos por un prefijo para dar nombre a estos archivos. Por ejemplo, si prefix
se establece en MyModel / myvariables / variables, entonces los archivos generados se encuentran debajo MyModel / myvariables y nombraron variables.data - * - * de-
Tenga en cuenta que este método podría alterar el gráfico subyacente si es la primera vez que una de sus sesiones se guarda, ver ERROR(/Graph#saverDef())
para más detalles.
Parámetros
prefijo | prefijo a los archivos variables para guardar |
---|