flusso tensoriale:: operazioni:: Ripristina V2

#include <io_ops.h>

Ripristina i tensori da un checkpoint V2.

Riepilogo

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

  • Questa Op prima tenta di trovare il file di indice V2 a cui punta il "prefisso" e, se trovato, procede a leggerlo come checkpoint V2;
  • Altrimenti viene richiamato il percorso di lettura V1. Non è consigliabile basarsi su questo comportamento, poiché la possibilità di ripiegare per leggere V1 potrebbe essere deprecata e infine rimossa.

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

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

Argomenti:

  • ambito: un oggetto ambito
  • prefisso: deve avere un singolo elemento. Il prefisso di un checkpoint V2.
  • nomi_tensore: forma {N}. I nomi dei tensori da ripristinare.
  • shape_and_slices: forma {N}. Le specifiche della sezione dei tensori da ripristinare. Le stringhe vuote indicano che sono tensori non partizionati.
  • tipi d: forma {N}. L'elenco dei dtype previsti per i tensori. Deve corrispondere a quelli memorizzati nel checkpoint.

Ritorna:

  • OutputList : forma {N}. I tensori ripristinati, 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