tf.raw_ops.SparseAccumulatorTakeGradient
Извлекает средний разреженный градиент из SparseConditionalAccumulator.
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