SendTPUEmbeddingGradients

classe finale publique SendTPUEmbeddingGradients

Effectue des mises à jour dégradées des tables d'intégration.

Méthodes publiques

SendTPUEmbeddingGradients statique
créer ( Scope scope, Iterable< Operand <Float>> entrées, Iterable< Operand <Float>> learningRates, String config)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SendTPUEmbeddingGradients.

Méthodes héritées

Méthodes publiques

public static SendTPUEmbeddingGradients créer (portée de portée , entrées Iterable < Operand <Float>>, Iterable < Operand <Float>> learningRates, configuration de chaîne)

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