tensorflow::ops::ScatterAdd
#include <state_ops.h>
Добавляет разреженные обновления к ссылке на переменную.
Краткое описание
Эта операция вычисляет
# Scalar indices ref[indices, ...] += updates[...] # Vector indices (for each i) ref[indices[i], ...] += updates[i, ...] # High rank indices (for each i, ..., j) ref[indices[i, ..., j], ...] += updates[i, ..., j, ...]
Эта операция выводит ref после завершения обновления. Это упрощает цепочку операций, которым нужно использовать исходное значение.
Дублирующиеся записи обрабатываются корректно: если несколько indices ссылаются на одно и то же место, их вклады суммируются.
Требуется updates.shape = indices.shape + ref.shape[1:] или updates.shape = [].
Аргументы:
- scope: Объект Scope
- ref: Должен быть из узла
Variable. - indices: Массив индексов в первом измерении
ref. - updates: Массив обновляемых значений для добавления к
ref.
Необязательные атрибуты (см. Attrs):
- use_locking: Если True, добавление будет защищено блокировкой; в противном случае поведение не определено, но может характеризоваться меньшей конкуренцией.
Возвращает:
-
Output: = То же, что иref. Возвращается для удобства операций, которые хотят использовать обновленные значения после завершения обновления.
| Конструкторы и деструкторы | |
|---|---|
ScatterAdd(const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates) | |
ScatterAdd(const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterAdd::Attrs & attrs) |
| Публичные атрибуты | |
|---|---|
operation | |
output_ref | |
| Публичные функции | |
|---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const | |
| Публичные статические функции | |
|---|---|
UseLocking(bool x) | |
| Структуры | |
|---|---|
| tensorflow::ops::ScatterAdd::Attrs | Необязательные параметры установки для ScatterAdd. |
Публичные атрибуты
operation
Operation operation
output_ref
::tensorflow::Output output_ref
Публичные функции
ScatterAdd
ScatterAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
ScatterAdd
ScatterAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterAdd::Attrs & attrs )
узел
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Публичные статические функции
UseLocking
Attrs UseLocking( bool x )
© 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/scatter-add