Driver per l'esecuzione Graph
.
Un'istanza Session
incapsula l'ambiente in cui vengono eseguite Operation
in un Graph
per calcolare Tensors
. Per esempio:
// 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
}
}
}
ATTENZIONE: una Session
possiede risorse che devono essere liberate esplicitamente invocando close()
.
Le istanze di una sessione sono thread-safe.
Classi nidificate
classe | Session.Run | Tensori di output e metadati ottenuti durante l'esecuzione di una sessione. | |
classe | Session.Runner | Esegui Operation e valuta Tensors . |
Costruttori pubblici
Sessione ( Grafico g, configurazione ConfigProto ) Costruisci una nuova sessione con il Graph associato e le opzioni di configurazione. |
Metodi pubblici
vuoto | vicino () Rilascia risorse associate alla sessione. |
vuoto | ripristino (prefisso stringa) Ripristina lo stato attuale delle variabili del grafico di questa sessione. |
vuoto | |
vuoto | esegui (String opName) Esegue un'operazione nel grafico con il nome dato. |
vuoto | eseguiInit () Esegui gli inizializzatori del grafico. |
Session.Runner | corridore () Crea un Runner per eseguire operazioni sui grafici e valutare i tensori. |
vuoto | salva (prefisso stringa) Salva lo stato attuale delle variabili del grafico di questa sessione. |
Metodi ereditati
Costruttori pubblici
Sessione pubblica ( Grafico g)
Sessione pubblica ( Grafico g, configurazione ConfigProto )
Costruisci una nuova sessione con il Graph
associato e le opzioni di configurazione.
Parametri
G | Il Graph su cui opererà la Sessione creata. |
---|---|
config | Parametri di configurazione per la sessione specificata come buffer del protocollo ConfigProto . |
Lancia
IllegalArgumentException | se la configurazione non è una serializzazione valida del buffer del protocollo ConfigProto. |
---|
Metodi pubblici
pubblico vuoto chiudi ()
Rilascia risorse associate alla sessione.
Si blocca finché non ci sono esecuzioni attive (chiamate run()
). Una Sessione non è utilizzabile dopo i rendimenti ravvicinati.
ripristino pubblico nullo (prefisso stringa)
Ripristina lo stato attuale delle variabili del grafico di questa sessione.
prefix
è il percorso in cui risiedono i file contenenti le variabili, seguito dal prefisso del nome file. Ad esempio, se prefix
è impostato su mymodel/myvariables/variables , i file vengono caricati da mymodel/myvariables e denominati variabili.data-*-of-*
Tieni presente che questo metodo potrebbe alterare il grafico sottostante se è la prima volta che una delle sue sessioni viene salvata, vedi ERROR(/Graph#saverDef())
per maggiori dettagli.
Parametri
prefisso | prefisso da cui ripristinare |
---|
esecuzione pubblica nulla ( Op op)
Esegue un'operazione nel grafico.
Questo metodo è equivalente a session.runner().addTarget(op).run()
.
Parametri
op | l'operazione da eseguire. |
---|
esecuzione pubblica void (String opName)
Esegue un'operazione nel grafico con il nome dato.
Questo metodo è equivalente a session.runner().addTarget(opName).run()
.
Parametri
opNome | nome dell'operazione da eseguire. |
---|
Lancia
IllegalArgumentException | se nel grafico non è possibile trovare alcuna operazione con quel nome |
---|
public void runInit ()
Esegui gli inizializzatori del grafico.
Questo metodo è equivalente a session.run(Ops.create(session.graph).init())
.
sessione pubblica.Runner corridore ()
Crea un Runner per eseguire operazioni sui grafici e valutare i tensori.
salvataggio pubblico nullo (prefisso stringa)
Salva lo stato attuale delle variabili del grafico di questa sessione.
prefix
è un percorso in cui verranno salvati i file contenenti le variabili stato, seguito da un prefisso per denominare questi file. Ad esempio, se prefix
è impostato su mymodel/myvariables/variables , i file generati verranno posizionati in mymodel/myvariables e denominati variable.data-*-of-*
Tieni presente che questo metodo potrebbe alterare il grafico sottostante se è la prima volta che una delle sue sessioni viene salvata, vedi ERROR(/Graph#saverDef())
per maggiori dettagli.
Parametri
prefisso | prefisso ai file variabili da salvare |
---|