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_signment=`[[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]`.
Metody publiczne
| Wyjście <T> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
| statyczny <T numer rozszerzenia> CrossReplicaSum <T> | |
| Wyjście <T> | wyjście () Suma wszystkich rozproszonych wejść. |
Metody dziedziczone
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 <liczba całkowita> 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