tf.raw_ops.SendTPUEmbeddingGradients
Выполняет обновление таблиц эмбеддингов с помощью градиентов.
tf.raw_ops.SendTPUEmbeddingGradients(
inputs, learning_rates, config, name=None
)
| Аргументы | |
|---|---|
inputs | Список как минимум из 1 объекта Tensor типа float32. TensorList градиентов, с помощью которых обновляются таблицы эмбеддингов. Этот аргумент имеет ту же длину и форму, что и возвращаемое значение RecvTPUEmbeddingActivations, но содержит градиенты функции потерь модели по отношению к активациям эмбеддингов. Таблицы эмбеддингов обновляются с помощью этих градиентов и оптимизатора, указанного в конфигурации встраивания TPU, переданной в tpu.initialize_system. |
learning_rates | Список объектов Tensor типа float32. TensorList скаляров 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
