classe finale publique CollectivePermute
Une opération pour permuter les tenseurs sur les instances TPU répliquées.
Chaque instance fournit sa propre entrée.
Par exemple, supposons qu'il y ait 4 instances TPU : `[A, B, C, D]`. Passer source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` obtient les sorties : `[D, A, B, C]`.
Méthodes publiques
| Sortie <T> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
| statique <T> CollectivePermute <T> | créer ( Portée de portée , Entrée Opérande <T>, Opérande <Integer> sourceTargetPairs) Méthode d'usine pour créer une classe encapsulant une nouvelle opération CollectivePermute. |
| Sortie <T> | sortir () L'entrée permutée. |
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 CollectivePermute <T> create ( Scope scope, Operand <T> input, Operand <Integer> sourceTargetPairs)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération CollectivePermute.
Paramètres
| portée | portée actuelle |
|---|---|
| saisir | L'entrée locale à permuter. Actuellement, ne prend en charge que float et bfloat16. |
| sourceTargetPaires | Un tenseur de forme [num_pairs, 2]. |
Retour
- une nouvelle instance de CollectivePermute