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.

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 > 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

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "NcclReduce"

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 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 ()