Spec-Zone.ru › TensorFlow

tf.raw_ops.LoadTPUEmbeddingFTRLParameters

Загрузка параметров эмбеддинга FTRL.

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

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

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

tf.compat.v1.raw_ops.LoadTPUEmbeddingFTRLParameters

tf.raw_ops.LoadTPUEmbeddingFTRLParameters(
    parameters,
    accumulators,
    linears,
    num_shards,
    shard_id,
    table_id=-1,
    table_name='',
    config='',
    name=None
)

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

Аргументы
parameters Tensor типа float32. Значение параметров, используемых в алгоритме оптимизации FTRL.
accumulators Tensor типа float32. Значение аккумуляторов, используемых в алгоритме оптимизации FTRL.
linears Tensor типа float32. Значение линейных параметров, используемых в алгоритме оптимизации FTRL.
num_shards int.
shard_id int.
table_id Необязательный int. Значение по умолчанию: -1.
table_name Необязательный string. Значение по умолчанию: "".
config Необязательный string. Значение по умолчанию: "".
name Имя операции (необязательно).
Возвращаемое значение
Созданная операция.

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API