classe finale publique CollectiveGatherV2
Accumule 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 | CollectiveGatherV2.Options | Attributs facultatifs pour CollectiveGatherV2 | |
Méthodes publiques
| Sortie <T> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
| CollectiveGatherV2.Options statique | communicationHint (String communicationHint) |
| statique <T étend le nombre> CollectiveGatherV2 <T> | |
| Sortie <T> | données () |
| CollectiveGatherV2.Options statique | isStateless (booléen isStateless) |
| CollectiveGatherV2.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 CollectiveGatherV2 <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 CollectiveGatherV2.
Paramètres
| portée | portée actuelle |
|---|---|
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de CollectiveGatherV2