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