tensorflow :: ops :: RestoreV2

#include <io_ops.h>

Restaura tensores de um ponto de verificação V2.

Resumo

Para compatibilidade com versões anteriores com o formato V1, este Op atualmente permite a restauração de um ponto de verificação V1 também:

  • Este Op primeiro tenta encontrar o arquivo de índice V2 apontado por "prefixo" e, se encontrado, prossegue para lê-lo como um ponto de verificação V2;
  • Caso contrário, o caminho de leitura V1 é invocado. Depender desse comportamento não é recomendado, pois a capacidade de retroceder para ler V1 pode ser descontinuada e eventualmente removida.

Por padrão, restaura os tensores nomeados por completo. Se o chamador deseja restaurar fatias específicas de tensores armazenados, "shape_and_slices" deve ser strings não vazias e correspondentemente bem formadas.

Os chamadores devem garantir que todos os tensores nomeados sejam de fato armazenados no ponto de verificação.

Argumentos:

  • escopo: um objeto Scope
  • prefixo: deve ter um único elemento. O prefixo de um ponto de verificação V2.
  • tensor_names: forma {N}. Os nomes dos tensores a serem restaurados.
  • shape_and_slices: shape {N}. As especificações de fatia dos tensores a serem restaurados. Strings vazias indicam que são tensores não particionados.
  • dtypes: forma {N}. A lista de dtype esperado para os tensores. Deve corresponder aos armazenados no ponto de verificação.

Retorna:

  • OutputList : forma {N}. Os tensores restaurados, cujas formas são lidas diretamente do ponto de verificação.

Construtores e Destruidores

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

Atributos públicos

operation
tensors

Funções públicas

operator[] (size_t index) const

Atributos públicos

Operação

Operation operation

tensores

::tensorflow::OutputList tensors

Funções públicas

RestoreV2

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

operador[]

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