tf.raw_ops.ConditionalAccumulator
Условный накопитель для агрегирования градиентов.
tf.raw_ops.ConditionalAccumulator(
dtype, shape, container='', shared_name='',
reduction_type='MEAN', name=None
)
Накопитель принимает градиенты, помеченные локальным шагом, больше или равным последнему известному глобальному шагу накопителя. Среднее значение можно извлечь из накопителя при условии накопления достаточного количества градиентов. Извлечение среднего значения автоматически сбрасывает агрегат в 0 и увеличивает глобальный шаг, записанный накопителем.
| Аргументы | |
|---|---|
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 типа mutable string. |
© 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/ConditionalAccumulator