AllToAll

publiczne zajęcia końcowe AllToAll

Opcja wymiany danych między replikami TPU.

W każdej replice dane wejściowe są dzielone na bloki „split_count” wzdłuż „split_dimension” i wysyłane do innych replik z określonym przypisaniem_grupy. Po otrzymaniu bloków `split_count` - 1 od innych replik, łączymy bloki wzdłuż `concat_dimension` jako wyniku.

Załóżmy na przykład, że istnieją 2 repliki TPU: replika 0 otrzymuje dane wejściowe: `[[A, B]]` replika 1 otrzymuje dane wejściowe: `[[C, D]]`

group_przypisanie=`[[0, 1]]` concat_dimension=0 split_dimension=1 split_count=2

wyjście repliki 0: `[[A], [C]]` wyjście repliki 1: `[[B], [D]]`

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 > AllToAll <T>	utwórz ( Zakres zakresu, Operand <T> wejście, Operand < TInt32 > groupAssignment, Long concatDimension, Long splitDimension, Long splitCount) Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację AllToAll.
Wyjście <T>	wyjście () Wymieniony wynik.

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: „AllToAll”

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 AllToAll <T> create ( Zakres zakresu, Operand <T> wejście, Operand < TInt32 > groupAssignment, Long concatDimension, Long splitDimension, Long splitCount)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację AllToAll.

Parametry

zakres	aktualny zakres
wejście	Lokalne wejście do sumy.
zadanie grupowe	Tensor int32 o kształcie [num_groups, num_replicas_per_group]. `przypisanie_grupy[i]` reprezentuje identyfikatory replik w i-tej podgrupie.
concatDimension	Numer wymiaru do połączenia.
podzielony wymiar	Numer wymiaru do podziału.
podzielna liczba	Liczba podziałów, liczba ta musi być równa wielkości podgrupy (group_przypisanie.get_shape()[1])

Zwroty

• nowa instancja AllToAll

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

Wymieniony wynik.