EnqueueTPUEmbeddingBatch

classe finale publique EnqueueTPUEmbeddingBatch

Une opération qui met en file d'attente une liste de tenseurs de lots d'entrée dans TPUEmbedding.

Une opération qui met en file d'attente une liste de tenseurs de lots d'entrée dans TPUEmbedding.

Classes imbriquées

classe Mettre en file d'attenteTPUEmbeddingBatch.Options Attributs facultatifs pour EnqueueTPUEmbeddingBatch

Méthodes publiques

EnqueueTPUEmbeddingBatch.Options statique
combinateurs (combineurs List<String>)
EnqueueTPUEmbeddingBatch statique
créer ( Scope scope, Iterable< Operand <String>> batch, Operand <String> modeOverride, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération EnqueueTPUEmbeddingBatch.
EnqueueTPUEmbeddingBatch.Options statique
appareilOrdinal (appareilOrdinal long)

Méthodes héritées

Méthodes publiques

Combineurs publics statiques EnqueueTPUEmbeddingBatch.Options (combineurs List<String>)

Paramètres
combinateurs Une liste de scalaires de chaînes, une pour chaque table d'intégration, qui spécifient comment normaliser les activations d'intégration après la sommation pondérée. Les combinateurs pris en charge sont « moyenne », « somme » ou « sqrtn ». Il n'est pas valide que la somme des poids soit égale à 0 pour « moyenne » ou que la somme des poids au carré soit égale à 0 pour « sqrtn ». Si les combinateurs ne sont pas transmis, la valeur par défaut est d'utiliser « somme » pour toutes les tables.

public static EnqueueTPUEmbeddingBatch create ( Scope scope, Iterable< Operand <String>> batch, Operand <String> modeOverride, Options... options)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération EnqueueTPUEmbeddingBatch.

Paramètres
portée portée actuelle
lot Une liste de tenseurs 1D, un pour chaque table d'intégration, contenant les entrées par lots codées en tant que protos dist_belief.SparseFeatures. Si le champ de poids dans le proto SparseFeatures n'est pas renseigné pour un ID, un poids de 1,0 est supposé.
modeOverride Entrée de chaîne qui remplace le mode spécifié dans TPUEmbeddingConfiguration. Les valeurs prises en charge sont {'unspecified', 'inference', 'training', 'backward_pass_only'}. Lorsqu'il est défini sur « non spécifié », le mode défini dans TPUEmbeddingConfiguration est utilisé, sinon mode_override est utilisé.
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de EnqueueTPUEmbeddingBatch

public static EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)

Paramètres
appareilOrdinal Le périphérique TPU à utiliser. Cela doit être -1 lorsque l'opération s'exécute sur un périphérique TPU et >= 0 lorsque l'opération s'exécute sur le périphérique CPU.