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
doSparseTensor
A, tamanho[nnz(A), ndims]
. - b_índices: 2-D. Os
indices
doSparseTensor
B, tamanho[nnz(B), ndims]
. - índices_soma: 2-D. Os
indices
da somaSparseTensor
, 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 )