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