NcclReduce

classe finale publique NcclReduce

Réduit `input` de `num_devices` en utilisant `reduction` à un seul périphérique.

Réduit `input` de `num_devices` en utilisant `reduction` à un seul périphérique.

Le graphique doit être construit de manière à ce que toutes les entrées aient une affectation de périphérique valide et que l'opération elle-même se voie attribuer l'un de ces périphériques.

input : L’entrée de la réduction. data : la valeur de la réduction sur tous les appareils `num_devices`. réduction : l’opération de réduction à effectuer.

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
statique <T étend le nombre> NcclReduce <T>
créer (portée de portée , entrée Iterable< Operand <T>>, réduction de chaîne)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération NcclReduce.
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 NcclReduce <T> create ( Scope scope, Iterable< Operand <T>> input, String réduction)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération NcclReduce.

Paramètres
portée portée actuelle
Retour
  • une nouvelle instance de NcclReduce

public Sortie <T> données ()