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.3/api_docs/python/tf/raw_ops/ConditionalAccumulator
