flujo tensor:: operaciones:: Restaurar
#include <io_ops.h>
Restaura un tensor a partir de archivos de puntos de control.
Resumen
Lee un tensor almacenado en uno o varios archivos. Si hay varios archivos (por ejemplo, porque un tensor se guardó como sectores), file_pattern
puede contener símbolos comodín ( *
y ?
) solo en la parte del nombre del archivo, no en la parte del directorio.
Si un file_pattern
coincide con varios archivos, preferred_shard
se puede utilizar para indicar en qué archivo es probable que se encuentre el tensor solicitado. Esta operación primero abrirá el archivo en el índice preferred_shard
en la lista de archivos coincidentes e intentará restaurar los tensores de ese archivo. Solo si no se encuentran algunos tensores o segmentos de tensor en ese primer archivo, el operador abre todos los archivos. Configurar preferred_shard
para que coincida con el valor pasado como entrada shard
de una operación Save
coincidente puede acelerar la restauración . Este atributo sólo afecta el rendimiento, no la corrección. El valor predeterminado -1 significa que los archivos se procesan en orden.
Consulte también RestoreSlice
.
Argumentos:
- alcance: un objeto de alcance
- file_pattern: Debe tener un solo elemento. El patrón de los archivos de los que leemos el tensor.
- tensor_name: debe tener un solo elemento. El nombre del tensor que se va a restaurar.
- dt: El tipo de tensor que se va a restaurar.
Atributos opcionales (ver Attrs
):
- fragmento_preferido: índice del archivo que se abrirá primero si varios archivos coinciden con
file_pattern
.
Devoluciones:
-
Output
: El tensor restaurado.
Constructores y destructores | |
---|---|
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) |
Atributos públicos | |
---|---|
operation | |
tensor |
Funciones públicas | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Funciones estáticas públicas | |
---|---|
PreferredShard (int64 x) |
estructuras | |
---|---|
tensorflow:: operaciones:: Restaurar:: Atributos | Configuradores de atributos opcionales para Restaurar . |
Atributos públicos
operación
Operation operation
tensor
::tensorflow::Output tensor
Funciones públicas
Restaurar
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt )
Restaurar
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs )
nodo
::tensorflow::Node * node() const
operador::tensorflow::Entrada
operator::tensorflow::Input() const
operador::tensorflow::Salida
operator::tensorflow::Output() const
Funciones estáticas públicas
Fragmento preferido
Attrs PreferredShard( int64 x )