tensoreflusso:: ops:: SparseAddGrad
#include <sparse_ops.h>
L'operatore gradiente per SparseAdd op.
Riepilogo
L'operazione SparseAdd calcola A + B, dove A, B e la somma sono tutti rappresentati come oggetti SparseTensor
. Questa operazione accetta il gradiente a monte rispetto ai valori non vuoti della somma e restituisce i gradienti rispetto ai valori non vuoti di A e B.
Argomenti:
- scope: un oggetto Scope
- backprop_val_grad: 1-D con forma
[nnz(sum)]
. Il gradiente rispetto ai valori non vuoti della somma. - a_indici: 2-D. Gli
indices
delloSparseTensor
A, dimensione[nnz(A), ndims]
. - b_indici: 2-D. Gli
indices
delloSparseTensor
B, dimensione[nnz(B), ndims]
. - somma_indici: 2-D. Gli
indices
della sommaSparseTensor
, size[nnz(sum), ndims]
.
Resi:
-
Output
a_val_grad: 1-D con forma[nnz(A)]
. Il gradiente rispetto ai valori non vuoti di A. -
Output
b_val_grad: 1-D con forma[nnz(B)]
. Il gradiente rispetto ai valori non vuoti di B.
Costruttori e distruttori | |
---|---|
SparseAddGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input backprop_val_grad, :: tensorflow::Input a_indices, :: tensorflow::Input b_indices, :: tensorflow::Input sum_indices) |
Attributi pubblici | |
---|---|
a_val_grad | |
b_val_grad | |
operation |
Attributi pubblici
a_val_grad
::tensorflow::Output a_val_grad
b_val_grad
::tensorflow::Output b_val_grad
operazione
Operation operation
Funzioni pubbliche
SparseAddGrad
SparseAddGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input backprop_val_grad, ::tensorflow::Input a_indices, ::tensorflow::Input b_indices, ::tensorflow::Input sum_indices )