CrossReplicaSum

publiczna klasa końcowa CrossReplicaSum

Opcja sumowania danych wejściowych z replikowanych instancji TPU.

Każda instancja dostarcza własne dane wejściowe.

Załóżmy na przykład, że istnieje 8 instancji TPU: `[A, B, C, D, E, F, G, H]`. Przekazanie group_ przypisania=`[[0,2,4,6],[1,3,5,7]]` ustawia `A, C, E, G` jako grupę 0 oraz `B, D, F, H` jako grupa 1. Otrzymujemy zatem wyjścia: `[A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+ G, B+D+F+H, A+C+E+G, B+D+F+H]`.

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 TNumber > CrossReplicaSum <T>	utwórz (zakres zakresu , wejście argumentu <T>, argument <TInt32> groupAssignment) Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację CrossReplicaSum.
Wyjście <T>	wyjście () Suma wszystkich rozproszonych wejść.

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

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 CrossReplicaSum <T> utwórz (zakres zakresu , argument wejściowy <T>, argument <TInt32> groupAssignment)

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

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.

Zwroty

• nowa instancja CrossReplicaSum

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

Suma wszystkich rozproszonych wejść.