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.qint16, tf.quint16, 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 типа изменяемый string. |
© 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/ConditionalAccumulator
