AllReduce

classe finale publique AllReduce

Réduit mutuellement plusieurs tenseurs de type et de forme identiques.

Classes imbriquées

classe AllReduce.Options Attributs facultatifs pour AllReduce

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 AllReduce.Options
communicationHint (String communicationHint)
statique <T étend TNumber > AllReduce <T>
créer ( Scope scope, Operand <T> input, Long groupSize, Long groupKey, Long instanceKey, String mergeOp, String finalOp, List<Long> subdivOffsets, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération AllReduce.
Sortie <T>
statique AllReduce.Options
timeoutSeconds (Flottant timeoutSeconds)
statique AllReduce.Options
waitFor (Liste<Long> waitFor)

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 : "CollectiveReduce"

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 statique AllReduce.Options communicationHint (String communicationHint)

public static AllReduce <T> create ( Scope scope, Operand <T> input, Long groupSize, Long groupKey, Long instanceKey, String mergeOp, String finalOp, List<Long> subdivOffsets, Options... options)

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

Paramètres
portée portée actuelle
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de AllReduce

public Sortie <T> données ()

public statique AllReduce.Options timeoutSeconds (Float timeoutSeconds)

public statique AllReduce.Options waitFor (List<Long> waitFor)