SendTPUEmbeddingGradients

publiczna klasa końcowa SendTPUEmbeddingGradients

Wykonuje aktualizacje gradientowe osadzanych tabel.

Metody publiczne

statyczne SendTPUEmbeddingGradients
utwórz (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.

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