тензорный поток:: опс:: РазреженныйДобавитьГрад

#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
)