tf.raw_ops.SparseConditionalAccumulator
Условный аккумулирующий агрегат для агрегирования разреженных градиентов.
tf.raw_ops.SparseConditionalAccumulator(
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 типа изменяемого 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/versions/r2.9/api_docs/python/tf/raw_ops/SparseConditionalAccumulator