tf.raw_ops.RetrieveTPUEmbeddingCenteredRMSPropParameters

Извлечение централизованных параметров RMSProp встраивания.

Псевдонимы

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

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

tf.compat.v1.raw_ops.RetrieveTPUEmbeddingCenteredRMSPropParameters

tf.raw_ops.RetrieveTPUEmbeddingCenteredRMSPropParameters(
    num_shards,
    shard_id,
    table_id=-1,
    table_name='',
    config='',
    name=None
)

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

Аргументы
`num_shards`	An `int`.
`shard_id`	An `int`.
`table_id`	Необязательный `int`. По умолчанию `-1`.
`table_name`	Необязательный `string`. По умолчанию `""`.
`config`	Необязательный `string`. По умолчанию `""`.
`name`	Имя операции (необязательно).

Возвращаемое значение
Кортеж объектов `Tensor` (parameters, ms, mom, mg).
`parameters`	A `Tensor` типа `float32`.
`ms`	A `Tensor` типа `float32`.
`mom`	A `Tensor` типа `float32`.
`mg`	A `Tensor` типа `float32`.

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