тензорный поток:: опс:: РазреженныйДобавитьГрад
#include <sparse_ops.h>
Оператор градиента для операции SparseAdd .
Краткое содержание
Операция SparseAdd вычисляет A + B, где A, B и сумма представлены как объекты SparseTensor
. Эта операция принимает восходящий градиент относительно непустых значений суммы и выводит градиенты относительно непустых значений A и B.
Аргументы:
- область: объект области.
- backprop_val_grad: 1-D с формой
[nnz(sum)]
. Градиент относительно непустых значений суммы. - a_indices: 2-D.
indices
SparseTensor
A, размер[nnz(A), ndims]
. - b_indices: 2-D.
indices
SparseTensor
B, размер[nnz(B), ndims]
. - сумма_индексов: 2-D.
indices
суммыSparseTensor
, размер[nnz(sum), ndims]
.
Возврат:
-
Output
a_val_grad: 1-D с формой[nnz(A)]
. Градиент по непустым значениям A. -
Output
b_val_grad: 1-D с формой[nnz(B)]
. Градиент относительно непустых значений B.
Конструкторы и деструкторы | |
---|---|
SparseAddGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input backprop_val_grad, :: tensorflow::Input a_indices, :: tensorflow::Input b_indices, :: tensorflow::Input sum_indices) |
Публичные атрибуты | |
---|---|
a_val_grad | |
b_val_grad | |
operation |
Публичные атрибуты
a_val_grad
::tensorflow::Output a_val_grad
b_val_grad
::tensorflow::Output b_val_grad
операция
Operation operation
Общественные функции
РазреженныйДобавитьГрад
SparseAddGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input backprop_val_grad, ::tensorflow::Input a_indices, ::tensorflow::Input b_indices, ::tensorflow::Input sum_indices )