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. |
---|