classe finale publique SendTPUEmbeddingGradients
Effectue des mises à jour dégradées des tables d'intégration.
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
SendTPUEmbeddingGradients statique |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Valeur constante : "SendTPUEmbeddingGradients"
Méthodes publiques
public static SendTPUEmbeddingGradients créer ( Scope scope, Iterable< Operand < TFloat32 >> entrées, Iterable< Operand < TFloat32 >> learningRates, String config)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SendTPUEmbeddingGradients.
Paramètres
portée | portée actuelle |
---|---|
contributions | Une TensorList de dégradés avec lesquels mettre à jour les tables d'intégration. Cet argument a la même longueur et la même forme que la valeur de retour de RecvTPUEmbeddingActivations, mais contient des gradients de perte du modèle par rapport aux activations d'intégration. Les tables d'intégration sont mises à jour à partir de ces dégradés via l'optimiseur spécifié dans la configuration d'intégration TPU donnée à tpu.initialize_system. |
Tarifs d'apprentissage | Une TensorList de scalaires float32, un pour chaque balise de taux d'apprentissage dynamique : voir les commentaires dans //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Plusieurs tables peuvent partager la même balise de taux d'apprentissage dynamique, comme spécifié dans la configuration. Si les taux d'apprentissage pour toutes les tables sont constants, cette liste doit être vide. |
configuration | Protocole TPUEmbeddingConfiguration sérialisé. |
Retour
- une nouvelle instance de SendTPUEmbeddingGradients