tf.raw_ops.LoadTPUEmbeddingCenteredRMSPropParameters

Загрузка параметров встраивания RMSProp с центрированием.

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

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

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

tf.compat.v1.raw_ops.LoadTPUEmbeddingCenteredRMSPropParameters

tf.raw_ops.LoadTPUEmbeddingCenteredRMSPropParameters(
    parameters, ms, mom, mg, num_shards, shard_id, table_id=-1,
    table_name='', config='', name=None
)

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

Аргументы
`parameters`	A `Tensor` типа `float32`. Значение параметров, используемых в алгоритме оптимизации RMSProp с центрированием.
`ms`	A `Tensor` типа `float32`. Значение ms, используемое в алгоритме оптимизации RMSProp с центрированием.
`mom`	A `Tensor` типа `float32`. Значение mom, используемое в алгоритме оптимизации RMSProp с центрированием.
`mg`	A `Tensor` типа `float32`. Значение mg, используемое в алгоритме оптимизации RMSProp с центрированием.
`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.4/api_docs/python/tf/raw_ops/LoadTPUEmbeddingCenteredRMSPropParameters