classe finale publique CollectiveReduceScatterV2
Réduit mutuellement plusieurs tenseurs de type et de forme identiques et disperse le résultat.
« 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 | 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> | données () |
| statique CollectiveReduceScatterV2.Options | isStateless (booléen isStateless) |
| 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 <T> create ( 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