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