tf.raw_ops.LoadTPUEmbeddingRMSPropParameters

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

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

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

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

tf.compat.v1.raw_ops.LoadTPUEmbeddingRMSPropParameters

tf.raw_ops.LoadTPUEmbeddingRMSPropParameters(
    parameters, ms, mom, 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.
`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/LoadTPUEmbeddingRMSPropParameters