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
ссылка_вывода
::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
оператор::tensorflow::Вход
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const
Публичные статические функции
ИспользоватьБлокировку
Attrs UseLocking( bool x )
© 2020 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.4/api_docs/cc/class/tensorflow/ops/scatter-add