RetrieveAllTPUEmbeddingParameters

classe final pública RetrieveAllTPUEmbeddingParameters

Um op que recupera os parâmetros de otimização da incorporação na memória do host.

Um op que recupera os parâmetros de otimização da incorporação na memória do host. Deve ser precedido por uma operação ConfigureTPUEmbeddingHost que define a configuração correta da tabela de incorporação. Por exemplo, este op é usado para recuperar parâmetros atualizados antes de salvar um ponto de verificação. Para Adagrad, auxiliar1 conterá os acumuladores após executar esta operação. Para SGD, todos os valores auxiliares* estarão vazios (tensores 0x0 para essa tabela). Para FTRL, auxiliar1 conterá os acumuladores e auxiliar2 conterá os termos lineares. Para ADAM, auxiliar1 conterá os momentos e auxiliar2 conterá as velocidades.

Métodos públicos

Lista< Saída <Float>>
auxiliar1 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o primeiro parâmetro de otimização auxiliar armazenado.
Lista< Saída <Float>>
auxiliar2 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o segundo parâmetro de otimização auxiliar armazenado.
Lista< Saída <Float>>
auxiliar3 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o terceiro parâmetro de otimização auxiliar armazenado.
Lista< Saída <Float>>
auxiliar4 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o quarto parâmetro de otimização auxiliar armazenado.
Lista< Saída <Float>>
auxiliar5 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o quinto parâmetro de otimização auxiliar armazenado.
Lista< Saída <Float>>
auxiliar6 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo os seis parâmetros de otimização auxiliares armazenados.
Lista< Saída <Float>>
auxiliar7 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o sétimo parâmetro de otimização auxiliar armazenado.
static RetrieveAllTPUEmbeddingParameters
create ( Escopo do escopo, Long NumTables, String config, Long numShards, Long shardId)
Método de fábrica para criar uma classe que envolve uma nova operação RetrieveAllTPUEmbeddingParameters.
Lista< Saída <Float>>
parâmetros ()
Uma lista de tensores, um para cada tabela de incorporação, contendo os parâmetros da tabela de incorporação armazenados.

Métodos Herdados

Métodos públicos

public List< Saída <Float>> auxiliar1 ()

Uma lista de tensores, um para cada tabela de incorporação, contendo o primeiro parâmetro de otimização auxiliar armazenado. Os elementos estão presentes na lista, mas têm tamanho zero, para parâmetros de otimização não utilizados (com base no algoritmo em uso para cada tabela).

public List< Saída <Float>> auxiliar2 ()

Uma lista de tensores, um para cada tabela de incorporação, contendo o segundo parâmetro de otimização auxiliar armazenado. Os elementos estão presentes na lista, mas têm tamanho zero, para parâmetros de otimização não utilizados (com base no algoritmo em uso para cada tabela).

public List< Saída <Float>> auxiliar3 ()

Uma lista de tensores, um para cada tabela de incorporação, contendo o terceiro parâmetro de otimização auxiliar armazenado. Os elementos estão presentes na lista, mas têm tamanho zero, para parâmetros de otimização não utilizados (com base no algoritmo em uso para cada tabela).

public List< Saída <Float>> auxiliar4 ()

Uma lista de tensores, um para cada tabela de incorporação, contendo o quarto parâmetro de otimização auxiliar armazenado. Os elementos estão presentes na lista, mas têm tamanho zero, para parâmetros de otimização não utilizados (com base no algoritmo em uso para cada tabela).

public List< Saída <Float>> auxiliar5 ()

Uma lista de tensores, um para cada tabela de incorporação, contendo o quinto parâmetro de otimização auxiliar armazenado. Os elementos estão presentes na lista, mas têm tamanho zero, para parâmetros de otimização não utilizados (com base no algoritmo em uso para cada tabela).

public List< Saída <Float>> auxiliar6 ()

Uma lista de tensores, um para cada tabela de incorporação, contendo os seis parâmetros de otimização auxiliares armazenados. Os elementos estão presentes na lista, mas têm tamanho zero, para parâmetros de otimização não utilizados (com base no algoritmo em uso para cada tabela).

public List< Saída <Float>> auxiliar7 ()

Uma lista de tensores, um para cada tabela de incorporação, contendo o sétimo parâmetro de otimização auxiliar armazenado. Os elementos estão presentes na lista, mas têm tamanho zero, para parâmetros de otimização não utilizados (com base no algoritmo em uso para cada tabela).

public static RetrieveAllTPUEmbeddingParameters create ( Escopo do escopo, Long NumTables, String config, Long numShards, Long shardId)

Método de fábrica para criar uma classe que envolve uma nova operação RetrieveAllTPUEmbeddingParameters.

Parâmetros
alcance escopo atual
NumTables O número de tabelas de incorporação.
configuração Um protocolo TPUEmbeddingConfiguration que descreve os parâmetros da tabela que estão sendo carregados, serializados em uma string.
numShards Número de estilhaços nos quais as tabelas de incorporação são divididas.
shardId Identificador do estilhaço para esta operação.
Devoluções
  • uma nova instância de RetrieveAllTPUEmbeddingParameters

public List< Output <Float>> parâmetros ()

Uma lista de tensores, um para cada tabela de incorporação, contendo os parâmetros da tabela de incorporação armazenados.