classe statique publique EnqueueTPUEmbeddingSparseBatch.Options
Attributs facultatifs pour EnqueueTPUEmbeddingSparseBatch
Méthodes publiques
| Mettre en file d'attenteTPUEmbeddingSparseBatch.Options | combinateurs (combineurs List<String>) |
| Mettre en file d'attenteTPUEmbeddingSparseBatch.Options | appareilOrdinal (appareilOrdinal long) |
Méthodes héritées
Méthodes publiques
Combineurs publics EnqueueTPUEmbeddingSparseBatch.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 EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (DeviceOrdinal long)
Paramètres
| appareilOrdinal | Le périphérique TPU à utiliser. Doit être >= 0 et inférieur au nombre de cœurs TPU dans la tâche sur laquelle le nœud est placé. |
|---|