classe statique publique EnqueueTPUEmbeddingBatch.Options
Attributs facultatifs pour EnqueueTPUEmbeddingBatch
Méthodes publiques
Mettre en file d'attenteTPUEmbeddingBatch.Options | combinateurs (combineurs List<String>) |
Mettre en file d'attenteTPUEmbeddingBatch.Options | appareilOrdinal (appareilOrdinal long) |
Méthodes héritées
Méthodes publiques
Combineurs publics 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 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. |
---|