tf.raw_ops.RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug

Получение параметров встраивания SGD с поддержкой отладки.

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

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

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

tf.compat.v1.raw_ops.RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug

tf.raw_ops.RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug(
    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, gradient_accumulators).
`parameters`	`Tensor` типа `float32`.
`gradient_accumulators`	`Tensor` типа `float32`.

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