CollectiveReduceV2

classe finale publique CollectiveReduceV2

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

« is_stateless » signifie que chaque opération n'a pas besoin de contrôler les dépendances avec d'autres opérations collectives. Dans ce cas, des clés uniques au moment de l'exécution (par exemple `instance_key`) doivent être utilisées pour distinguer les groupes collectifs.

Classes imbriquées

classe CollectiveReduceV2.Options Attributs facultatifs pour CollectiveReduceV2

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
statique CollectiveReduceV2.Options
communicationHint (String communicationHint)
statique <T étend le nombre> CollectiveReduceV2 <T>
créer ( Scope scope, Operand <T> input, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanceKey, Iterable< Operand <?>> orderingToken, String mergeOp, String finalOp, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération CollectiveReduceV2.
Sortie <T>
statique CollectiveReduceV2.Options
isStateless (booléen isStateless)
statique CollectiveReduceV2.Options
maxSubdivsPerDevice (Long maxSubdivsPerDevice)
statique CollectiveReduceV2.Options
timeoutSeconds (Flottant timeoutSeconds)

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

public static CollectiveReduceV2 <T> créer ( Scope scope, Operand <T> input, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanceKey, Iterable< Operand <?>> orderingToken, String mergeOp, String finalOp, Options... options)

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

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

public Sortie <T> données ()

public static CollectiveReduceV2.Options isStateless (booléen isStateless)

public statique CollectiveReduceV2.Options maxSubdivsPerDevice (Long maxSubdivsPerDevice)

public static CollectiveReduceV2.Options timeoutSeconds (Float timeoutSeconds)