Une opération qui récupère les paramètres d'optimisation de l'intégration dans la mémoire hôte.
Une opération qui récupère les paramètres d'optimisation de l'intégration dans la mémoire hôte. Doit être précédé d'une opération ConfigureTPUEmbeddingHost qui configure la configuration correcte de la table d'intégration. Par exemple, cette opération est utilisée pour récupérer les paramètres mis à jour avant d'enregistrer un point de contrôle. Pour Adagrad, auxiliaire1 contiendra les accumulateurs après avoir exécuté cette opération. Pour SGD, toutes les valeurs auxiliaires* seront vides (tenseurs 0x0 pour cette table). Pour FTRL, auxiliaire1 contiendra les accumulateurs et auxiliaire2 contiendra les termes linéaires. Pour ADAM, auxiliaire1 contiendra l’impulsion et auxiliaire2 contiendra les vitesses.
Méthodes publiques
| Liste< Sortie <Float>> | auxiliaire1 () Une liste de tenseurs, un pour chaque table d'intégration, contenant le premier paramètre d'optimisation auxiliaire stocké. |
| Liste< Sortie <Float>> | auxiliaire2 () Une liste de tenseurs, un pour chaque table d'intégration, contenant le deuxième paramètre d'optimisation auxiliaire stocké. |
| Liste< Sortie <Float>> | auxiliaire3 () Une liste de tenseurs, un pour chaque table d'intégration, contenant le troisième paramètre d'optimisation auxiliaire stocké. |
| Liste< Sortie <Float>> | auxiliaire4 () Une liste de tenseurs, un pour chaque table d'intégration, contenant le quatrième paramètre d'optimisation auxiliaire stocké. |
| Liste< Sortie <Float>> | auxiliaire5 () Une liste de tenseurs, un pour chaque table d'intégration, contenant le cinquième paramètre d'optimisation auxiliaire stocké. |
| Liste< Sortie <Float>> | auxiliaire6 () Une liste de tenseurs, un pour chaque table d'intégration, contenant les six paramètres d'optimisation auxiliaires stockés. |
| Liste< Sortie <Float>> | auxiliaire7 () Une liste de tenseurs, un pour chaque table d'intégration, contenant le septième paramètre d'optimisation auxiliaire stocké. |
| statique RetrieveAllTPUEmbeddingParameters | |
| Liste< Sortie <Float>> | paramètres () Une liste de tenseurs, un pour chaque table d'intégration, contenant les paramètres de la table d'intégration stockés. |
Méthodes héritées
Méthodes publiques
public List< Sortie <Float>> auxiliaire1 ()
Une liste de tenseurs, un pour chaque table d'intégration, contenant le premier paramètre d'optimisation auxiliaire stocké. Les éléments sont présents dans la liste, mais ont une taille nulle, pour les paramètres d'optimisation inutilisés (basés sur l'algorithme utilisé pour chaque table).
liste publique< Sortie <Float>> auxiliaire2 ()
Une liste de tenseurs, un pour chaque table d'intégration, contenant le deuxième paramètre d'optimisation auxiliaire stocké. Les éléments sont présents dans la liste, mais ont une taille nulle, pour les paramètres d'optimisation inutilisés (basés sur l'algorithme utilisé pour chaque table).
liste publique < Sortie <Float>> auxiliaire3 ()
Une liste de tenseurs, un pour chaque table d'intégration, contenant le troisième paramètre d'optimisation auxiliaire stocké. Les éléments sont présents dans la liste, mais ont une taille nulle, pour les paramètres d'optimisation inutilisés (basés sur l'algorithme utilisé pour chaque table).
liste publique < Sortie <Float>> auxiliaire4 ()
Une liste de tenseurs, un pour chaque table d'intégration, contenant le quatrième paramètre d'optimisation auxiliaire stocké. Les éléments sont présents dans la liste, mais ont une taille nulle, pour les paramètres d'optimisation inutilisés (basés sur l'algorithme utilisé pour chaque table).
liste publique < Sortie <Float>> auxiliaire5 ()
Une liste de tenseurs, un pour chaque table d'intégration, contenant le cinquième paramètre d'optimisation auxiliaire stocké. Les éléments sont présents dans la liste, mais ont une taille nulle, pour les paramètres d'optimisation inutilisés (basés sur l'algorithme utilisé pour chaque table).
liste publique < Sortie <Float>> auxiliaire6 ()
Une liste de tenseurs, un pour chaque table d'intégration, contenant les six paramètres d'optimisation auxiliaires stockés. Les éléments sont présents dans la liste, mais ont une taille nulle, pour les paramètres d'optimisation inutilisés (basés sur l'algorithme utilisé pour chaque table).
liste publique < Sortie <Float>> auxiliaire7 ()
Une liste de tenseurs, un pour chaque table d'intégration, contenant le septième paramètre d'optimisation auxiliaire stocké. Les éléments sont présents dans la liste, mais ont une taille nulle, pour les paramètres d'optimisation inutilisés (basés sur l'algorithme utilisé pour chaque table).
public static RetrieveAllTPUEmbeddingParameters créer ( scope scope, Long NumTables, String config, Long numShards, Long shardId)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération RetrieveAllTPUEmbeddingParameters.
Paramètres
| portée | portée actuelle |
|---|---|
| Nombre de tables | Le nombre de tables d'intégration. |
| configuration | Un prototype TPUEmbeddingConfiguration décrivant les paramètres de table en cours de chargement, sérialisés en une chaîne. |
| nombre de fragments | Nombre de fragments dans lesquels les tables d'intégration sont divisées. |
| ID de fragment | Identifiant du fragment pour cette opération. |
Retour
- une nouvelle instance de RetrieveAllTPUEmbeddingParameters
liste publique < Paramètres de sortie <Float>> ()
Une liste de tenseurs, un pour chaque table d'intégration, contenant les paramètres de la table d'intégration stockés.