tf.raw_ops.ResourceConditionalAccumulator
Условный накопитель для агрегирования градиентов.
tf.raw_ops.ResourceConditionalAccumulator(
dtype, shape, container='', shared_name='',
reduction_type='MEAN', name=None
)
Накопитель принимает градиенты, помеченные local_step, большим или равным последнему известному глобальному шагу, известному накопителю. Среднее значение можно извлечь из накопителя, если было накоплено достаточно градиентов. Извлечение среднего значения автоматически сбрасывает агрегат в 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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/ResourceConditionalAccumulator
