publiczna klasa końcowa SendTPUEmbeddingGradients
Wykonuje aktualizacje gradientowe osadzanych tabel.
Stałe
| Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
| statyczne SendTPUEmbeddingGradients |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Wartość stała: „SendTPUEmbeddingGradients”
Metody publiczne
public static SendTPUEmbeddingGradients create ( Zakres zakresu , Iterable< Operand < TFloat32 >> wejścia, Iterable< Operand < TFloat32 >> learningRates, String config)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację SendTPUEmbeddingGradients.
Parametry
| zakres | aktualny zakres |
|---|---|
| wejścia | TensorList gradientów, za pomocą których można aktualizować tabele osadzania. Argument ten ma taką samą długość i kształt jak wartość zwracana przez RecvTPUEmbeddingActivations, ale zawiera gradienty straty modelu w odniesieniu do aktywacji osadzania. Tabele osadzania są aktualizowane na podstawie tych gradientów za pomocą optymalizatora określonego w konfiguracji osadzania TPU przekazanej do tpu.initialize_system. |
| stawki za naukę | Lista TensorList skalarów typu float32, po jednym dla każdego znacznika dynamicznej szybkości uczenia się: zobacz komentarze w pliku // Third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Wiele tabel może współużytkować ten sam znacznik dynamicznego współczynnika uczenia się, jak określono w konfiguracji. Jeśli szybkości uczenia się dla wszystkich tabel są stałe, lista ta powinna być pusta. |
| konfiguracja | Serializowany proto konfiguracji TPUEmbeddingConfiguration. |
Zwroty
- nowa instancja SendTPUEmbeddingGradients