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 .

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)

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
)