publiczna klasa końcowa SendTPUEmbeddingGradients
Wykonuje aktualizacje gradientowe osadzanych tabel.
Metody publiczne
| statyczne SendTPUEmbeddingGradients |
Metody dziedziczone
Metody publiczne
public static SendTPUEmbeddingGradients create (zakres zakresu , wejścia Iterable< Operand <Float>>, Iterable< Operand <Float>> learningRates, konfiguracja ciągu)
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