тензорный поток:: опс:: 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 = []
.
Аргументы:
- область: объект области.
- ref: Должно быть из узла
Variable
. - indexes: тензор индексов в первом измерении
ref
. - обновления: тензор обновленных значений для добавления в
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
выходной_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
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const
Публичные статические функции
Использование блокировки
Attrs UseLocking( bool x )