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> | |
| Sortie <T> | données () |
| 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 <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