ОтправитьТПУЭмбеддингГрадиентс

публичный финальный класс SendTPUEmbeddingGradients

Выполняет градиентное обновление таблиц внедрения.

Публичные методы

статический SendTPUEmbeddingGradients
create (область области , входы Iterable< Operand <Float>>, Iterable< Operand <Float>> LearningRates, строковая конфигурация)
Фабричный метод для создания класса, обертывающего новую операцию SendTPUEmbeddingGradients.

Унаследованные методы

Публичные методы

public static SendTPUEmbeddingGradients create (область видимости , Iterable< входы Operand <Float>>, Iterable< Operand <Float>> LearningRates, строковая конфигурация)

Фабричный метод для создания класса, обертывающего новую операцию SendTPUEmbeddingGradients.

Параметры
объем текущий объем
входы TensorList градиентов, с помощью которых можно обновить таблицы внедрения. Этот аргумент имеет ту же длину и форму, что и возвращаемое значение RecvTPUEmbeddingActivations, но содержит градиенты потерь модели по отношению к активации внедрения. Таблицы внедрения обновляются на основе этих градиентов с помощью оптимизатора, указанного в конфигурации внедрения TPU, указанной в tpu.initialize_system.
обучениеЦены TensorList скаляров float32, по одному для каждого тега динамической скорости обучения: см. комментарии в // Third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Несколько таблиц могут использовать один и тот же тег динамической скорости обучения, указанный в конфигурации. Если скорость обучения для всех таблиц постоянна, этот список должен быть пустым.
конфигурация Сериализованный прототип TPUEmbeddingConfiguration.
Возврат
  • новый экземпляр SendTPUEmbeddingGradients