EnqueueTPUEmbeddingBatch.Options

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

De la classe java.lang.Object

booléen	est égal à (Objet arg0)
Classe finale<?>	obtenirClasse ()
int	Code de hachage ()
vide final	notifier ()
vide final	notifierTous ()
Chaîne	àChaîne ()
vide final	attendre (long arg0, int arg1)
vide final	attendez (long arg0)
vide final	attendez ()

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.

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.