RetrieveAllTPUEmbeddingParameters

classe final pública RetrieveAllTPUEmbeddingParameters

Uma operação que recupera parâmetros de otimização da incorporação na memória do host.

Uma operação que recupera 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, esta operação é usada 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 auxiliar de otimização armazenado.
Lista< Saída <Float>>
auxiliar2 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o segundo parâmetro auxiliar de otimização armazenado.
Lista< Saída <Float>>
auxiliar3 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o terceiro parâmetro auxiliar de otimização armazenado.
Lista< Saída <Float>>
auxiliar4 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o quarto parâmetro auxiliar de otimização armazenado.
Lista< Saída <Float>>
auxiliar5 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o quinto parâmetro auxiliar de otimização armazenado.
Lista< Saída <Float>>
auxiliar6 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo os seis parâmetros auxiliares de otimização armazenados.
Lista< Saída <Float>>
auxiliar7 ()
Uma lista de tensores, um para cada tabela de incorporação, contendo o sétimo parâmetro auxiliar de otimização armazenado.
estática RetrieveAllTPUEmbeddingParameters
criar (escopo de 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

Lista pública< Saída <Float>> auxiliar1 ()

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

Lista pública< Saída <Float>> auxiliar2 ()

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

Lista pública< Saída <Float>> auxiliar3 ()

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

Lista pública< Saída <Float>> auxiliar4 ()

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

Lista pública< Saída <Float>> auxiliar5 ()

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

Lista pública< Saída <Float>> auxiliar6 ()

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

Lista pública< Saída <Float>> auxiliar7 ()

Uma lista de tensores, um para cada tabela de incorporação, contendo o sétimo parâmetro auxiliar de otimização armazenado. Os elementos estão presentes na lista, mas possuem 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 de 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
escopo escopo atual
NumTables O número de tabelas de incorporação.
configuração Um proto TPUEmbeddingConfiguration que descreve os parâmetros da tabela que estão sendo carregados, serializados em uma string.
numShards Número de fragmentos nos quais as tabelas de incorporação são divididas.
fragmentoId Identificador do fragmento para esta operação.
Devoluções
  • uma nova instância de RetrieveAllTPUEmbeddingParameters

Lista pública< 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.