тензорный поток :: ops :: SparseAddGrad

#include <sparse_ops.h>

Оператор градиента для SparseAdd op.

Резюме

Операция SparseAdd вычисляет A + B, где A, B и сумма представлены как объекты SparseTensor . Эта операция принимает восходящий градиент по непустым значениям суммы и выводит градиенты по непустым значениям A и B.

Аргументы:

  • scope: объект Scope
  • backprop_val_grad: 1-D с формой [nnz(sum)] . Градиент относительно непустых значений суммы.
  • a_indices: 2-D. indices SparseTensor A, size [nnz(A), ndims] .
  • b_indices: 2-D. indices SparseTensor B, size [nnz(B), ndims] .
  • sum_indices: 2-D. indices суммы SparseTensor , size [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

 SparseAddGrad(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input backprop_val_grad,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input b_indices,
  ::tensorflow::Input sum_indices
)