fluxo tensor:: ops:: SparseAddGrad

#include <sparse_ops.h>

O operador gradiente para a operação SparseAdd .

Resumo

A operação SparseAdd calcula A + B, onde A, B e a soma são todos representados como objetos SparseTensor . Esta operação pega o gradiente upstream em relação aos valores não vazios da soma e gera os gradientes em relação aos valores não vazios de A e B.

Argumentos:

  • escopo: um objeto Escopo
  • backprop_val_grad: 1-D com forma [nnz(sum)] . O gradiente em relação aos valores não vazios da soma.
  • a_índices: 2-D. Os indices do SparseTensor A, tamanho [nnz(A), ndims] .
  • b_índices: 2-D. Os indices do SparseTensor B, tamanho [nnz(B), ndims] .
  • índices_soma: 2-D. Os indices da soma SparseTensor , size [nnz(sum), ndims] .

Retorna:

  • Output a_val_grad: 1-D com forma [nnz(A)] . O gradiente em relação aos valores não vazios de A.
  • Output b_val_grad: 1-D com forma [nnz(B)] . O gradiente em relação aos valores não vazios de B.

Construtores e Destruidores

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

Atributos públicos

a_val_grad
b_val_grad
operation

Atributos públicos

a_val_grad

::tensorflow::Output a_val_grad

b_val_grad

::tensorflow::Output b_val_grad

operação

Operation operation

Funções públicas

SparseAddGrad

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