flux tensoriel : : opérations : : Restaurer
#include <io_ops.h>Restaure un tenseur à partir des fichiers de points de contrôle.
Résumé
Lit un tenseur stocké dans un ou plusieurs fichiers. S'il existe plusieurs fichiers (par exemple parce qu'un tenseur a été enregistré sous forme de tranches), file_pattern peut contenir des symboles génériques ( * et ? ) uniquement dans la partie nom de fichier, pas dans la partie répertoire.
Si un file_pattern correspond à plusieurs fichiers, preferred_shard peut être utilisé pour indiquer dans quel fichier le tenseur demandé est susceptible de se trouver. Cette opération ouvrira d'abord le fichier à l'index preferred_shard dans la liste des fichiers correspondants et tentera de restaurer les tenseurs à partir de ce fichier. Seulement si certains tenseurs ou tranches de tenseurs ne sont pas trouvés dans ce premier fichier, alors l'Op ouvre tous les fichiers. Définir preferred_shard pour qu'il corresponde à la valeur transmise comme entrée shard d'une opération Save correspondante peut accélérer Restore . Cet attribut affecte uniquement les performances, pas l'exactitude. La valeur par défaut -1 signifie que les fichiers sont traités dans l'ordre.
Voir également RestoreSlice .
Arguments :
- scope : un objet Scope
- file_pattern : doit avoir un seul élément. Le modèle des fichiers à partir desquels nous lisons le tenseur.
- tensor_name : doit avoir un seul élément. Le nom du tenseur à restaurer.
- dt : Le type du tenseur à restaurer.
Attributs facultatifs (voir Attrs ) :
- Preferred_shard : index du fichier à ouvrir en premier si plusieurs fichiers correspondent à
file_pattern.
Retours :
-
Output: Le tenseur restauré.
Constructeurs et Destructeurs | |
|---|---|
Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt) | |
Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs) |
Attributs publics | |
|---|---|
operation | |
tensor | |
Fonctions publiques | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Fonctions statiques publiques | |
|---|---|
PreferredShard (int64 x) | |
Structures | |
|---|---|
| tensorflow :: ops :: Restauration :: Attrs | Setters d'attributs facultatifs pour Restore . |
Attributs publics
opération
Operation operation
tenseur
::tensorflow::Output tensor
Fonctions publiques
Restaurer
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt )
Restaurer
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs )
nœud
::tensorflow::Node * node() const
opérateur :: tensorflow :: Entrée
operator::tensorflow::Input() const
opérateur :: tensorflow :: Sortie
operator::tensorflow::Output() const
Fonctions statiques publiques
Fragment préféré
Attrs PreferredShard( int64 x )