CollectiveAllToAllV2

classe finale publique CollectiveAllToAllV2

Échange 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 CollectifAllToAllV2.Options Attributs facultatifs pour CollectiveAllToAllV2

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
CollectiveAllToAllV2.Options statique
communicationHint (String communicationHint)
statique <T étend le numéro> CollectiveAllToAllV2 <T>
créer ( Scope scope, Operand <T> input, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanceKey, Iterable< Operand <?>> orderingToken, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération CollectiveAllToAllV2.
Sortie <T>
CollectiveAllToAllV2.Options statique
isStateless (booléen isStateless)
CollectiveAllToAllV2.Options statique
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 CollectiveAllToAllV2.Options communicationHint (String communicationHint)

public static CollectiveAllToAllV2 <T> create ( Scope scope, Operand <T> input, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanceKey, Iterable< Operand <?>> orderingToken, Options... options)

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

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

public Sortie <T> données ()

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

public static CollectiveAllToAllV2.Options timeoutSeconds (Float timeoutSeconds)