NcclAllReduce

classe finale publique NcclAllReduce

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>
créer (portée de portée , entrée opérande <T>, réduction de chaîne, numDevices longs, chaîne sharedName)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération NcclAllReduce.
Sortie <T>

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

public Sortie <T> données ()