tensorflow :: opérations :: Restaurer

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é.

Attributs publics

Operation operation

::tensorflow::Output tensor

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs PreferredShard(
  int64 x
)