tensorflow :: opérations :: Restaurer

#include <io_ops.h>

Restaure un tenseur à partir des fichiers de point de contrôle.

Résumé

Lit un tenseur stocké dans un ou plusieurs fichiers. S'il y a 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 indication dans quel fichier le tenseur 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 tenseur ne sont pas trouvés dans ce premier fichier, alors l'Op ouvre tous les fichiers. La définition de preferred_shard pour qu'elle corresponde à la valeur transmise comme entrée de shard d'une opération de Save correspondante peut accélérer la restauration . 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 motif des fichiers à partir desquels nous lisons le tenseur.
  • nom_tensor: 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 .

Retour:

  • 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)

Structs

tensorflow :: ops :: Restore :: Attrs

Paramètres d'attributs facultatifs pour la restauration .

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 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const 

Fonctions statiques publiques

PreferredShard

Attrs PreferredShard(
  int64 x
)