Pilote pour l’exécution Graph
.
Une instance Session
encapsule l'environnement dans lequel les Operation
dans un Graph
sont exécutées pour calculer Tensors
. Par exemple:
// 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
}
}
}
AVERTISSEMENT : une Session
possède des ressources qui doivent être explicitement libérées en appelant close()
.
Les instances d'une session sont thread-safe.
Classes imbriquées
classe | Session.Exécuter | Tenseurs de sortie et métadonnées obtenues lors de l'exécution d'une session. | |
classe | Session.Runner | Exécutez Operation s et évaluez Tensors . |
Constructeurs Publics
Session ( Graphique g, config ConfigProto ) Construisez une nouvelle session avec les options Graph et de configuration associées. |
Méthodes publiques
vide | fermer () Libérer les ressources associées à la session. |
vide | restaurer (préfixe de chaîne) Restaurer l'état réel des variables du graphique de cette session. |
vide | |
vide | exécuter (String opName) Exécute une opération dans le graphique portant le nom donné. |
vide | exécuterInit () Exécutez les initialiseurs du graphique. |
Session.Runner | coureur () Créez un Runner pour exécuter des opérations graphiques et évaluer les Tensors. |
vide | enregistrer (préfixe de chaîne) Enregistre l'état réel des variables du graphique de cette session. |
Méthodes héritées
Constructeurs Publics
Séance publique ( Graphique g)
Session publique ( Graphique g, configuration ConfigProto )
Construisez une nouvelle session avec les options Graph
et de configuration associées.
Paramètres
g | Le Graph sur lequel la session créée fonctionnera. |
---|---|
configuration | Paramètres de configuration pour la session spécifiée comme tampon de protocole ConfigProto . |
Jetés
IllegalArgumentException | si la configuration n'est pas une sérialisation valide du tampon de protocole ConfigProto. |
---|
Méthodes publiques
public vide fermer ()
Libérer les ressources associées à la session.
Bloque jusqu'à ce qu'il n'y ait plus d'exécutions actives (appels run()
). Une session n'est pas utilisable après les retours de clôture.
restauration publique vide (préfixe de chaîne)
Restaurer l'état réel des variables du graphique de cette session.
prefix
est le chemin où se trouvent les fichiers contenant l'état des variables, suivi du préfixe du nom de fichier. Par exemple, si prefix
est défini sur mymodel/myvariables/variables , alors les fichiers sont chargés depuis mymodel/myvariables et nommés variables.data-*-of-*
Notez que cette méthode peut modifier le graphique sous-jacent si c'est la première fois qu'une de ses sessions est enregistrée, voir ERROR(/Graph#saverDef())
pour plus de détails.
Paramètres
préfixe | préfixe à partir duquel restaurer |
---|
course vide publique ( Op op)
Exécute une opération dans le graphique.
Cette méthode est équivalente à session.runner().addTarget(op).run()
.
Paramètres
op | l’opération à exécuter. |
---|
exécution publique vide (String opName)
Exécute une opération dans le graphique portant le nom donné.
Cette méthode est équivalente à session.runner().addTarget(opName).run()
.
Paramètres
NomOp | nom de l'opération à exécuter. |
---|
Jetés
IllegalArgumentException | si aucune opération de ce nom ne peut être trouvée dans le graphique |
---|
public vide runInit ()
Exécutez les initialiseurs du graphique.
Cette méthode est équivalente à session.run(Ops.create(session.graph).init())
.
public Session.Runner coureur ()
Créez un Runner pour exécuter des opérations graphiques et évaluer les Tensors.
sauvegarde publique vide (préfixe de chaîne)
Enregistre l'état réel des variables du graphique de cette session.
prefix
est un chemin où seront enregistrés les fichiers contenant l'état des variables, suivi d'un préfixe pour nommer ces fichiers. Par exemple, si prefix
est défini sur mymodel/myvariables/variables , alors les fichiers générés seront situés sous mymodel/myvariables et nommés variables.data-*-of-*
Notez que cette méthode peut modifier le graphique sous-jacent si c'est la première fois qu'une de ses sessions est enregistrée, voir ERROR(/Graph#saverDef())
pour plus de détails.
Paramètres
préfixe | préfixe aux fichiers de variables à enregistrer |
---|