tf.raw_ops.AccumulatorTakeGradient

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

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

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

Дополнительные сведения см. в руководстве по миграции.

tf.compat.v1.raw_ops.AccumulatorTakeGradient

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

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

Аргументы
`handle`	`Tensor` типа изменяемого `string`. Дескриптор аккумулятора.
`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` типа `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/AccumulatorTakeGradient