tf.raw_ops.AccumulatorTakeGradient

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

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

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

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

tf.compat.v1.raw_ops.AccumulatorTakeGradient

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

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

Аргументы
`handle`	A `Tensor` типа mutable `string`. Дескриптор накопителя.
`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.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/versions/r2.9/api_docs/python/tf/raw_ops/AccumulatorTakeGradient