tf.raw_ops.LoadTPUEmbeddingProximalAdagradParameters

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

Псевдонимы

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

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

tf.compat.v1.raw_ops.LoadTPUEmbeddingProximalAdagradParameters

tf.raw_ops.LoadTPUEmbeddingProximalAdagradParameters(
    parameters,
    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.
`num_shards`	An `int`.
`shard_id`	An `int`.
`table_id`	Необязательный `int`. По умолчанию `-1`.
`table_name`	Необязательный `string`. По умолчанию `""`.
`config`	Необязательный `string`. По умолчанию `""`.
`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/LoadTPUEmbeddingProximalAdagradParameters