flux tensoriel : : opérations : : RestaurerV2

#include <io_ops.h>

Restaure les tenseurs à partir d'un point de contrôle V2.

Résumé

Pour une compatibilité ascendante avec le format V1, cette opération permet actuellement également la restauration à partir d'un point de contrôle V1 :

  • Cette opération tente d'abord de trouver le fichier d'index V2 pointé par "préfixe", et s'il est trouvé, procédez à sa lecture comme point de contrôle V2 ;
  • Sinon, le chemin de lecture V1 est invoqué. Il n'est pas recommandé de s'appuyer sur ce comportement, car la possibilité de revenir à la lecture de la V1 pourrait être obsolète, voire supprimée.

Par défaut, restaure intégralement les tenseurs nommés. Si l'appelant souhaite restaurer des tranches spécifiques de tenseurs stockés, "shape_and_slices" doit être des chaînes non vides et donc bien formées.

Les appelants doivent s'assurer que tous les tenseurs nommés sont bien stockés dans le point de contrôle.

Arguments :

  • scope : un objet Scope
  • préfixe : doit avoir un seul élément. Le préfixe d'un point de contrôle V2.
  • tensor_names : forme {N}. Les noms des tenseurs à restaurer.
  • shape_and_slices : forme {N}. Les spécifications de tranche des tenseurs à restaurer. Les chaînes vides indiquent qu'il s'agit de tenseurs non partitionnés.
  • types : forme {N}. La liste des types attendus pour les tenseurs. Doit correspondre à ceux stockés dans le point de contrôle.

Retours :

  • OutputList : forme {N}. Les tenseurs restaurés, dont les formes sont lues directement depuis le point de contrôle.

Constructeurs et Destructeurs

RestoreV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input prefix, :: tensorflow::Input tensor_names, :: tensorflow::Input shape_and_slices, const DataTypeSlice & dtypes)

Attributs publics

operation
tensors

Fonctions publiques

operator[] (size_t index) const

Attributs publics

opération

Operation operation

tenseurs

::tensorflow::OutputList tensors

Fonctions publiques

RestaurerV2

 RestoreV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input prefix,
  ::tensorflow::Input tensor_names,
  ::tensorflow::Input shape_and_slices,
  const DataTypeSlice & dtypes
)

opérateur[]

::tensorflow::Output operator[](
  size_t index
) const