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

Metody publiczne

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

Metody dziedziczone

Z klasy org.tensorflow.op.PrimitiveOp

końcowa wartość logiczna	równa się (obiekt obiektu)
końcowy wew	hashCode ()
Operacja	op () Zwraca podstawową Operation
ostatni ciąg	doString ()

Z klasy java.lang.Object

wartość logiczna	równa się (Obiekt arg0)
ostatnia klasa<?>	pobierzKlasę ()
wew	hashCode ()
ostateczna pustka	powiadomić ()
ostateczna pustka	powiadom wszystkich ()
Strunowy	doString ()
ostateczna pustka	czekaj (długi arg0, int arg1)
ostateczna pustka	czekaj (długi arg0)
ostateczna pustka	Czekać ()

Z interfejsu org.tensorflow.Operand

streszczenie Wyjście <T>

jako wyjście () Zwraca symboliczny uchwyt tensora.

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

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

Permutowane wejście.