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