Opcja umożliwiająca permutację tensorów w replikowanych instancjach TPU.
Każda instancja dostarcza własne dane wejściowe.
Załóżmy na przykład, że istnieją 4 instancje TPU: `[A, B, C, D]`. Przekazanie source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` daje wyniki: `[D, A, B, C]`.
Metody publiczne
Wyjście <T> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
statyczny <T> Przejście zbiorowe <T> | |
Wyjście <T> | wyjście () Permutowane wejście. |
Metody dziedziczone
Metody publiczne
publiczne wyjście <T> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static CollectivePermute <T> create (zakres zakresu , argument wejściowy <T>, argument <liczba całkowita> sourceTargetPairs)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację CollectivePermute.
Parametry
zakres | aktualny zakres |
---|---|
wejście | Lokalne wejście, które ma zostać permutowane. Obecnie obsługuje tylko float i bfloat16. |
sourceTargetPairs | Tensor o kształcie [num_pairs, 2]. |
Zwroty
- nowa instancja CollectivePermute