tf.raw_ops.ResourceConditionalAccumulator
Условный сумматор для агрегирования градиентов.
tf.raw_ops.ResourceConditionalAccumulator(
dtype,
shape,
container='',
shared_name='',
reduction_type='MEAN',
name=None
)
Сумматор принимает градиенты, помеченные локальным шагом, большим или равным последнему глобальному шагу, известному сумматору. Среднее значение может быть извлечено из сумматора при условии накопления достаточного количества градиентов. Извлечение среднего значения автоматически сбрасывает агрегат в 0 и увеличивает глобальный шаг, записанный сумматором. Это ресурсная версия ConditionalAccumulator, которая будет работать в TF2.0 с tf.cond версии 2.
| Аргументы | |
|---|---|
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. Тип накапливаемого значения. |
shape | tf.TensorShape или список ints. Форма значений, может быть [], в этом случае форма неизвестна. |
container | Необязательный string. По умолчанию "". Если не пустой, этот сумматор помещается в указанный контейнер. В противном случае используется контейнер по умолчанию. |
shared_name | Необязательный string. По умолчанию "". Если не пустой, этот сумматор будет совместно использоваться под заданным именем в нескольких сессиях. |
reduction_type | Необязательный string из: "MEAN", "SUM". По умолчанию "MEAN". |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Tensor типа resource. |
© 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/ResourceConditionalAccumulator
