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

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ść.