tensorflow::ops::RestoreV2
#include <io_ops.h>Ripristina i tensori da un checkpoint V2.
Riepilogo
Per garantire la compatibilità con il formato V1, questa operazione consente attualmente anche il ripristino da un checkpoint V1:
- Questa operazione tenta innanzitutto di trovare il file indice V2 indicato da "prefix" e, se trovato, procede alla sua lettura come checkpoint V2;
- Altrimenti viene richiamato il percorso di lettura V1. Fare affidamento su questo comportamento non è consigliabile, poiché la possibilità di ripiegare sulla lettura V1 potrebbe essere deprecata e infine rimossa.
Per impostazione predefinita, ripristina completamente i tensori specificati. Se il chiamante desidera ripristinare porzioni specifiche dei tensori memorizzati, "shape_and_slices" deve essere una stringa non vuota e, di conseguenza, ben formata.
Chi effettua la chiamata deve assicurarsi che tutti i tensori specificati siano effettivamente memorizzati nel checkpoint.
Argomentazioni:
- ambito: un oggetto Scope
- prefisso: Deve contenere un singolo elemento. Il prefisso di un checkpoint V2.
- tensor_names: forma {N}. I nomi dei tensori da ripristinare.
- shape_and_slices: forma {N}. Le specifiche delle sezioni dei tensori da ripristinare. Le stringhe vuote indicano che si tratta di tensori non partizionati.
- dtypes: forma {N}. L'elenco dei tipi di dati previsti per i tensori. Devono corrispondere a quelli memorizzati nel checkpoint.
Resi:
-
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
RestoreV2
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