SendTPUEmbeddingGradients

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

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