flux tensoriel : : opérations : : Restaurer

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

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
)