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.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 типа mutable 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/SparseConditionalAccumulator
