tf.raw_ops.LoadTPUEmbeddingAdagradMomentumParameters

Загружает параметры эмбеддингов Adagrad Momentum.

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

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.LoadTPUEmbeddingAdagradMomentumParameters

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

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

Аргументы
`parameters`	`Tensor` типа `float32`. Значение параметров, используемых алгоритмом оптимизации Adagrad Momentum.
`accumulators`	`Tensor` типа `float32`. Значение аккумуляторов, используемых алгоритмом оптимизации Adagrad Momentum.
`momenta`	`Tensor` типа `float32`. Значение моментов, используемых алгоритмом оптимизации Adagrad Momentum.
`num_shards`	`int`.
`shard_id`	`int`.
`table_id`	Необязательный `int`. По умолчанию — `-1`.
`table_name`	Необязательный `string`. По умолчанию — `""`.
`config`	Необязательный `string`. По умолчанию — `""`.
`name`	Имя операции (необязательно).

Возвращает
Созданный объект Operation.

© 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/LoadTPUEmbeddingAdagradMomentumParameters