tf.raw_ops.ResourceAccumulatorTakeGradient
Извлекает средний градиент в заданном ConditionalAccumulator.
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