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.
Méthodes publiques
| Sortie <T> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
| statique <T étend le nombre> NcclAllReduce <T> | |
| Sortie <T> | donné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 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