Spec-Zone.ru › TensorFlow 2.3

tf.raw_ops.LoadTPUEmbeddingAdagradParametersGradAccumDebug

Загрузка параметров встраивания Adagrad с поддержкой отладки.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.raw_ops.LoadTPUEmbeddingAdagradParametersGradAccumDebug

tf.raw_ops.LoadTPUEmbeddingAdagradParametersGradAccumDebug(
    parameters, accumulators, gradient_accumulators, num_shards, shard_id,
    table_id=-1, table_name='', config='', name=None
)

Операция, которая загружает параметры оптимизации в HBM для встраивания. Должна предшествовать операции ConfigureTPUEmbeddingHost, которая настраивает правильную конфигурацию таблицы встраивания. Например, эта операция используется для установки параметров, загруженных из контрольной точки перед выполнением цикла обучения.

Аргументы
parameters A Tensor типа float32. Значение параметров, используемых в алгоритме оптимизации Adagrad.
accumulators A Tensor типа float32. Значение аккумуляторов, используемых в алгоритме оптимизации Adagrad.
gradient_accumulators A Tensor типа float32. Значение gradient_accumulators, используемых в алгоритме оптимизации Adagrad.
num_shards An int.
shard_id An int.
table_id Необязательный int. По умолчанию -1.
table_name Необязательный string. По умолчанию "".
config Необязательный string. По умолчанию "".
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/LoadTPUEmbeddingAdagradParametersGradAccumDebug

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API