Une opération pour additionner les entrées sur les instances TPU répliquées.
Chaque instance fournit sa propre entrée.
Par exemple, supposons qu'il y ait 8 instances TPU : `[A, B, C, D, E, F, G, H]`. Passer group_assignment=`[[0,2,4,6],[1,3,5,7]]` définit `A, C, E, G` comme groupe 0 et `B, D, F, H` comme groupe 1. Ainsi nous obtenons les sorties : `[A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+ G, B+D+F+H, A+C+E+G, B+D+F+H]`.
Constantes
| Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
| Sortie <T> | comme Sortie () Renvoie le handle symbolique du tenseur. |
| statique <T étend TNumber > CrossReplicaSum <T> | |
| Sortie <T> | sortir () La somme de toutes les entrées distribuées. |
Méthodes héritées
| 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 () |
| résuméExécutionEnvironnement | env () Renvoie l'environnement d'exécution dans lequel cette opération a été créée. |
| opération abstraite |
| sortie abstraite <T> | comme Sortie () Renvoie le handle symbolique du tenseur. |
| résumé T | asTensor () Renvoie le tenseur à cet opérande. |
| Forme abstraite | |
| Classe abstraite<T> | taper () Renvoie le type de tenseur de cet opérande |
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
sortie publique <T> asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static CrossReplicaSum <T> créer ( Scope scope, Operand <T> input, Operand < TInt32 > groupAssignment)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération CrossReplicaSum.
Paramètres
| portée | portée actuelle |
|---|---|
| saisir | L'entrée locale à la somme. |
| affectation de groupe | Un tenseur int32 de forme [num_groups, num_replicas_per_group]. `group_assignment[i]` représente les identifiants de réplique dans le ième sous-groupe. |
Retour
- une nouvelle instance de CrossReplicaSum