публичный финальный класс SendTPUEmbeddingGradients
Выполняет градиентное обновление таблиц внедрения.
Публичные методы
статический 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