tensoreflusso:: ops:: Ripristina V2

#include <io_ops.h>

Ripristina i tensori da un checkpoint V2.

Riepilogo

Per compatibilità con le versioni precedenti con il formato V1, questa operazione attualmente consente il ripristino anche da un checkpoint V1:

  • Questo Op tenta innanzitutto di trovare il file indice V2 indicato da "prefix" e, se trovato, procede a leggerlo come checkpoint V2;
  • Altrimenti viene richiamato il percorso di lettura V1. Non è consigliabile fare affidamento su questo comportamento, poiché la possibilità di eseguire il fallback per leggere la V1 potrebbe essere deprecata ed eventualmente rimossa.

Per impostazione predefinita, ripristina completamente i tensori nominati. Se il chiamante desidera ripristinare porzioni specifiche di tensori memorizzati, "shape_and_slices" dovrebbe essere stringhe non vuote e corrispondentemente ben formate.

I chiamanti devono assicurarsi che tutti i tensori nominati siano effettivamente archiviati nel checkpoint.

Argomenti:

  • scope: un oggetto Scope
  • prefisso: deve avere un singolo elemento. Il prefisso di un checkpoint V2.
  • tensor_names: forma {N}. I nomi dei tensori da ripristinare.
  • forma_e_fette: forma {N}. Le specifiche della sezione dei tensori da ripristinare. Le stringhe vuote indicano che sono tensori non partizionati.
  • dtypes: forma {N}. L'elenco dei dtype attesi per i tensori. Deve corrispondere a quelli memorizzati nel checkpoint.

Ritorna:

  • OutputList : forma {N}. I tensori restaurati, le cui forme vengono lette direttamente dal checkpoint.

Costruttori e distruttori

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

Attributi pubblici

operation
tensors

Funzioni pubbliche

operator[] (size_t index) const

Attributi pubblici

operazione

Operation operation

tensori

::tensorflow::OutputList tensors

Funzioni pubbliche

Ripristina V2

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

operatore[]

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