tf.raw_ops.LoadTPUEmbeddingMDLAdagradLightParameters

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

Псевдонимы

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

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

tf.compat.v1.raw_ops.LoadTPUEmbeddingMDLAdagradLightParameters

tf.raw_ops.LoadTPUEmbeddingMDLAdagradLightParameters(
    parameters,
    accumulators,
    weights,
    benefits,
    num_shards,
    shard_id,
    table_id=-1,
    table_name='',
    config='',
    name=None
)

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

Аргументы
`parameters`	`Tensor` типа `float32`. Значение параметров, используемых алгоритмом оптимизации MDL Adagrad Light.
`accumulators`	`Tensor` типа `float32`. Значение аккумуляторов, используемых алгоритмом оптимизации MDL Adagrad Light.
`weights`	`Tensor` типа `float32`. Значение весов, используемых алгоритмом оптимизации MDL Adagrad Light.
`benefits`	`Tensor` типа `float32`. Значение преимуществ, используемых алгоритмом оптимизации MDL Adagrad Light.
`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/LoadTPUEmbeddingMDLAdagradLightParameters