tf.raw_ops.ResourceConditionalAccumulator
Условный аккумулятор для накопления градиентов.
tf.raw_ops.ResourceConditionalAccumulator(
dtype,
shape,
container='',
shared_name='',
reduction_type='MEAN',
name=None
)
Аккумулятор принимает градиенты, помеченные значением local_step, которое больше или равно последнему известному аккумулятору значению global_step. Среднее значение можно извлечь из аккумулятора, если накоплено достаточное количество градиентов. При извлечении среднего накопленное значение автоматически сбрасывается до 0, а значение global_step, зарегистрированное аккумулятором, увеличивается. Это ресурсная версия ConditionalAccumulator, которая работает в TF2.0 с версией 2 tf.cond.
| Аргументы | |
|---|---|
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: тип tf.DType. Тип накапливаемого значения. |
shape | ints или список значений tf.TensorShape. Форма значений; может быть [], в этом случае форма неизвестна. |
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/api_docs/python/tf/raw_ops/ResourceConditionalAccumulator
