tf.raw_ops.SendTPUEmbeddingGradients
Выполняет обновление градиентов таблиц встраивания.
tf.raw_ops.SendTPUEmbeddingGradients(
inputs, learning_rates, config, name=None
)
| Аргументы | |
|---|---|
inputs | Список, содержащий как минимум 1 объект типа float32. Список тензоров градиентов, используемых для обновления таблиц встраивания. Этот аргумент имеет такую же длину и формы, как и результат RecvTPUEmbeddingActivations, но содержит градиенты потери модели относительно активаций встраивания. Таблицы встраивания обновляются из этих градиентов с помощью оптимизатора, указанного в конфигурации TPU встраивания, переданной в tpu.initialize_system. |
learning_rates | Список объектов типа float32. Список тензоров float32, каждый из которых представляет собой скаляр, соответствующий метке динамической скорости обучения (см. комментарии в //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto). Несколько таблиц могут использовать одну и ту же метку динамической скорости обучения, как указано в конфигурации. Если скорости обучения для всех таблиц являются постоянными, этот список должен быть пустым. |
config | string. Сериализованный протокол TPUEmbeddingConfiguration. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/SendTPUEmbeddingGradients
