CollectivePermute

publiczna klasa końcowa CollectivePermute

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]`.

Stałe

Strunowy OP_NAME Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczny <T rozszerza TType > CollectivePermute <T>
utwórz (zakres zakresu , argument wejściowy <T>, argument <TInt32> sourceTargetPairs)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację CollectivePermute.
Wyjście <T>
wyjście ()
Permutowane wejście.

Metody dziedziczone

Stałe

publiczny statyczny końcowy ciąg znaków OP_NAME

Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Wartość stała: „ZbiorowePermute”

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 <TInt32> 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

publiczne wyjście <T> wyjście ()

Permutowane wejście.