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

Z klasy org.tensorflow.op.RawOp

końcowa wartość logiczna	równa się (obiekt obiektu)
końcowy wew	hashCode ()
Operacja	op () Zwróć tę jednostkę obliczeniową jako pojedynczą 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.op.Op

abstrakcyjne środowisko wykonania	środowisko () Zwróć środowisko wykonawcze, w którym utworzono tę operację.
abstrakcyjna operacja	op () Zwróć tę jednostkę obliczeniową jako pojedynczą Operation .

Z interfejsu org.tensorflow.Operand

streszczenie Wyjście <T>	jako wyjście () Zwraca symboliczny uchwyt tensora.
streszczenie T	asTensor () Zwraca tensor tego operandu.
abstrakcyjny kształt	kształt () Zwraca (prawdopodobnie częściowo znany) kształt tensora, do którego odnosi się dane Output tego operandu.
Klasa abstrakcyjna<T>	typ () Zwraca typ tensora tego operandu

Z interfejsu org.tensorflow.ndarray.Shaped

streszczenie wew	ranga ()
abstrakcyjny kształt	kształt ()
abstrakcyjne, długie	rozmiar () Oblicza i zwraca całkowity rozmiar tego kontenera w liczbie wartości.

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.