Génère un tenseur contenant la réduction sur tous les tenseurs d'entrée.
Génère un tenseur contenant la réduction sur tous les tenseurs d'entrée transmis aux opérations dans le même « nom_partagé ».
Le graphique doit être construit de sorte que si une opération s'exécute avec la valeur de nom_partagé `c`, alors les opérations `num_devices` s'exécuteront avec la valeur de nom_partagé `c`. Si vous ne le faites pas, l’exécution du graphique ne pourra pas se terminer.
input : l'entrée des données de réduction : la valeur de la réduction sur tous les appareils `num_devices`. réduction : l’opération de réduction à effectuer. num_devices : le nombre d'appareils participant à cette réduction. shared_name : identifiant partagé entre les opérations de la même réduction.
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 > NcclAllReduce <T> | |
| Sortie <T> | données () |
Méthodes héritées
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 NcclAllReduce <T> créer (portée de portée , entrée d'opérande <T>, réduction de chaîne, numDevices longs, nom partagé de chaîne)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération NcclAllReduce.
Paramètres
| portée | portée actuelle |
|---|
Retour
- une nouvelle instance de NcclAllReduce