Una operación que recupera parámetros de optimización al incrustarlos en la memoria del host.
Una operación que recupera parámetros de optimización al incrustarlos en la memoria del host. Debe estar precedido por una operación ConfigureTPUEmbeddingHost que establezca la configuración correcta de la tabla de incrustación. Por ejemplo, esta operación se utiliza para recuperar parámetros actualizados antes de guardar un punto de control. Para Adagrad, auxiliar1 contendrá los acumuladores después de ejecutar esta operación. Para SGD, todos los valores auxiliares* estarán vacíos (tensores 0x0 para esa tabla). Para FTRL, auxiliar1 contendrá los acumuladores y auxiliar2 contendrá los términos lineales. Para ADAM, auxiliar1 contendrá los momentos y auxiliar2 contendrá las velocidades.
Métodos públicos
| Lista< Salida <Flotante>> | auxiliar1 () Una lista de tensores, uno para cada tabla de incrustación, que contiene el primer parámetro de optimización auxiliar almacenado. |
| Lista< Salida <Flotante>> | auxiliar2 () Una lista de tensores, uno para cada tabla de incrustación, que contiene el segundo parámetro de optimización auxiliar almacenado. |
| Lista< Salida <Flotante>> | auxiliar3 () Una lista de tensores, uno para cada tabla de incrustación, que contiene el tercer parámetro de optimización auxiliar almacenado. |
| Lista< Salida <Flotante>> | auxiliar4 () Una lista de tensores, uno para cada tabla de incrustación, que contiene el cuarto parámetro de optimización auxiliar almacenado. |
| Lista< Salida <Flotante>> | auxiliar5 () Una lista de tensores, uno para cada tabla de incrustación, que contiene el quinto parámetro de optimización auxiliar almacenado. |
| Lista< Salida <Flotante>> | auxiliar6 () Una lista de tensores, uno para cada tabla de incrustación, que contiene los seis parámetros de optimización auxiliares almacenados. |
| Lista< Salida <Flotante>> | auxiliar7 () Una lista de tensores, uno para cada tabla de incrustación, que contiene el séptimo parámetro de optimización auxiliar almacenado. |
| Recuperar estáticamente todos los parámetros de integración de TPUUE | |
| Lista< Salida <Flotante>> | parámetros () Una lista de tensores, uno para cada tabla de incrustación, que contiene los parámetros almacenados de la tabla de incrustación. |
Métodos heredados
Métodos públicos
Lista pública< Salida <Float>> auxiliar1 ()
Una lista de tensores, uno para cada tabla de incrustación, que contiene el primer parámetro de optimización auxiliar almacenado. Los elementos están presentes en la lista, pero tienen tamaño cero, para los parámetros de optimización no utilizados (según el algoritmo utilizado para cada tabla).
Lista pública< Salida <Flotante>> auxiliar2 ()
Una lista de tensores, uno para cada tabla de incrustación, que contiene el segundo parámetro de optimización auxiliar almacenado. Los elementos están presentes en la lista, pero tienen tamaño cero, para los parámetros de optimización no utilizados (según el algoritmo utilizado para cada tabla).
Lista pública< Salida <Float>> auxiliar3 ()
Una lista de tensores, uno para cada tabla de incrustación, que contiene el tercer parámetro de optimización auxiliar almacenado. Los elementos están presentes en la lista, pero tienen tamaño cero, para los parámetros de optimización no utilizados (según el algoritmo utilizado para cada tabla).
Lista pública< Salida <Float>> auxiliar4 ()
Una lista de tensores, uno para cada tabla de incrustación, que contiene el cuarto parámetro de optimización auxiliar almacenado. Los elementos están presentes en la lista, pero tienen tamaño cero, para los parámetros de optimización no utilizados (según el algoritmo utilizado para cada tabla).
Lista pública< Salida <Float>> auxiliar5 ()
Una lista de tensores, uno para cada tabla de incrustación, que contiene el quinto parámetro de optimización auxiliar almacenado. Los elementos están presentes en la lista, pero tienen tamaño cero, para los parámetros de optimización no utilizados (según el algoritmo utilizado para cada tabla).
Lista pública< Salida <Float>> auxiliar6 ()
Una lista de tensores, uno para cada tabla de incrustación, que contiene los seis parámetros de optimización auxiliares almacenados. Los elementos están presentes en la lista, pero tienen tamaño cero, para los parámetros de optimización no utilizados (según el algoritmo utilizado para cada tabla).
Lista pública< Salida <Float>> auxiliar7 ()
Una lista de tensores, uno para cada tabla de incrustación, que contiene el séptimo parámetro de optimización auxiliar almacenado. Los elementos están presentes en la lista, pero tienen tamaño cero, para los parámetros de optimización no utilizados (según el algoritmo utilizado para cada tabla).
creación pública estática de RetrieveAllTPUEmbeddingParameters ( alcance de alcance , Long NumTables, configuración de cadena, Long numShards, Long shardId)
Método de fábrica para crear una clase que envuelve una nueva operación RetrieveAllTPUEmbeddingParameters.
Parámetros
| alcance | alcance actual |
|---|---|
| Tablas numéricas | El número de tablas de incrustación. |
| configuración | Un protocolo TPUEmbeddingConfiguration que describe los parámetros de la tabla que se cargan y se serializan en una cadena. |
| número de fragmentos | Número de fragmentos en los que se dividen las tablas de incrustación. |
| ID de fragmento | Identificador del fragmento para esta operación. |
Devoluciones
- una nueva instancia de RetrieveAllTPUEmbeddingParameters
Lista pública < Salida <Float>> parámetros ()
Una lista de tensores, uno para cada tabla de incrustación, que contiene los parámetros almacenados de la tabla de incrustación.