Spec-Zone.ru › TensorFlow C++ 2.4

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
Operation
output_ref
::tensorflow::Output
Публичные функции
node() const
::tensorflow::Node *
operator::tensorflow::Input() const
operator::tensorflow::Output() const
Публичные статические функции
UseLocking(bool x)
Attrs
Структуры
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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API