tf.raw_ops.RetrieveTPUEmbeddingMomentumParameters

Получает параметры эмбеддингов Momentum.

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

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

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

tf.compat.v1.raw_ops.RetrieveTPUEmbeddingMomentumParameters

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

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

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

Возвращает
Кортеж из объектов `Tensor` (параметры, импульсы).
`parameters`	Объект `Tensor` типа `float32`.
`momenta`	Объект `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/RetrieveTPUEmbeddingMomentumParameters