Spec-Zone.ru › TensorFlow

tf.raw_ops.SparseAccumulatorTakeGradient

Извлекает средний разреженный градиент из SparseConditionalAccumulator.

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

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

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

tf.compat.v1.raw_ops.SparseAccumulatorTakeGradient

tf.raw_ops.SparseAccumulatorTakeGradient(
    handle, num_required, dtype, name=None
)

Операция будет блокироваться, пока не будет накоплено достаточное количество градиентов (то есть больше num_required). Если аккумулятор уже агрегировал больше градиентов, чем num_required, он вернет их среднее значение. Также автоматически увеличивает записанное значение global_step в аккумуляторе на 1 и сбрасывает агрегат в 0.

Аргументы
handle Tensor типа изменяемого string. Дескриптор SparseConditionalAccumulator.
num_required Tensor типа int32. Количество градиентов, необходимое для возврата агрегата.
dtype tf.DType из: tf.float32, tf.float64, tf.int32, tf.uint8, tf.int16, tf.int8, tf.complex64, tf.int64, tf.qint8, tf.quint8, tf.qint32, tf.bfloat16, tf.qint16, tf.quint16, tf.uint16, tf.complex128, tf.half, tf.uint32, tf.uint64. Тип данных накопленных градиентов. Должен соответствовать типу аккумулятора.
name Имя операции (необязательно).
Возвращает
Кортеж объектов Tensor (индексы, значения, форма).
indices Tensor типа int64.
values Tensor типа dtype.
shape Tensor типа int64.

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

Spec-Zone.ru

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