CollectiveReduceScatterV2

classe finale publique CollectiveReduceScatterV2

Réduit mutuellement plusieurs tenseurs de type et de forme identiques et disperse le résultat.

Classes imbriquées

classe CollectiveReduceScatterV2.Options Attributs facultatifs pour CollectiveReduceScatterV2

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
statique CollectiveReduceScatterV2.Options
communicationHint (String communicationHint)
statique <T étend le nombre> CollectiveReduceScatterV2 <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 CollectiveReduceScatterV2.
Sortie <T>
statique CollectiveReduceScatterV2.Options
maxSubdivsPerDevice (Long maxSubdivsPerDevice)
statique CollectiveReduceScatterV2.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 CollectiveReduceScatterV2.Options communicationHint (String communicationHint)

public static CollectiveReduceScatterV2 <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 CollectiveReduceScatterV2.

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

public Sortie <T> données ()

public statique CollectiveReduceScatterV2.Options maxSubdivsPerDevice (Long maxSubdivsPerDevice)

public static CollectiveReduceScatterV2.Options timeoutSeconds (Float timeoutSeconds)