CrossReplicaSum

classe finale publique CrossReplicaSum

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

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
statique <T étend le nombre> CrossReplicaSum <T>
créer ( Scope scope, Operand <T> input, Operand <Integer> groupAssignment)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération CrossReplicaSum.
Sortie <T>
sortir ()
La somme de toutes les entrées distribuées.

Méthodes héritées

Méthodes publiques

sortie publique <T> asOutput ()

Renvoie le handle symbolique d'un 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 <Integer> 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

sortie publique <T> sortie ()

La somme de toutes les entrées distribuées.