tensorflow::ops::SparseAccumulatorTakeGradient
#include <data_flow_ops.h>
Извлекает усреднённый разреженный градиент из SparseConditionalAccumulator.
Краткое описание
Операция блокируется до тех пор, пока не будет накоплено достаточно градиентов (то есть больше, чем num_required). Если аккумулятор уже накопил больше градиентов, чем num_required, операция возвращает их усреднённое значение. Также автоматически увеличивает на 1 записанный в аккумуляторе global_step и сбрасывает агрегат до 0.
Аргументы:
- scope: объект Scope
- handle: дескриптор SparseConditionalAccumulator.
- num_required: количество градиентов, необходимое для возврата агрегата.
- dtype: тип данных накопленных градиентов. Должен соответствовать типу аккумулятора.
Возвращает:
-
Outputindices: индексы усреднённых накопленных разреженных градиентов. -
Outputvalues: значения усреднённых накопленных разреженных градиентов. -
Outputshape: форма усреднённых накопленных разреженных градиентов.
| Конструкторы и деструкторы | |
|---|---|
SparseAccumulatorTakeGradient(const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_required, DataType dtype) |
| Публичные атрибуты | |
|---|---|
indices | |
operation | |
shape | |
values | |
Публичные атрибуты
indices
::tensorflow::Output indices
operation
Operation operation
shape
::tensorflow::Output shape
values
::tensorflow::Output values
Публичные функции
SparseAccumulatorTakeGradient
SparseAccumulatorTakeGradient( const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_required, DataType dtype )
© 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/cc/class/tensorflow/ops/sparse-accumulator-take-gradient