Spec-Zone.ru › TensorFlow

tf.raw_ops.ResourceAccumulatorTakeGradient

Извлекает средний градиент в заданном ConditionalAccumulator.

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

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

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

tf.compat.v1.raw_ops.ResourceAccumulatorTakeGradient

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

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

Аргументы
handle A Tensor типа resource. Дескриптор активатора.
num_required A 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 Имя операции (необязательно).
Возвращаемое значение
A Tensor типа dtype.

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

Spec-Zone.ru

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