flux tensoriel : : opérations : : SparseAddGrad
#include <sparse_ops.h>
L'opérateur de dégradé pour l'opération SparseAdd .
Résumé
L'opération SparseAdd calcule A + B, où A, B et la somme sont tous représentés sous forme d'objets SparseTensor
. Cette opération prend en compte le gradient en amont par rapport aux valeurs non vides de la somme et génère les gradients par rapport aux valeurs non vides de A et B.
Arguments :
- scope : un objet Scope
- backprop_val_grad : 1-D avec forme
[nnz(sum)]
. Le gradient par rapport aux valeurs non vides de la somme. - a_indices : 2-D. Les
indices
duSparseTensor
A, size[nnz(A), ndims]
. - b_indices : 2-D. Les
indices
duSparseTensor
B, size[nnz(B), ndims]
. - sum_indices : 2-D. Les
indices
de la sommeSparseTensor
, size[nnz(sum), ndims]
.
Retours :
-
Output
a_val_grad : 1-D avec forme[nnz(A)]
. Le gradient par rapport aux valeurs non vides de A. -
Output
b_val_grad : 1-D avec forme[nnz(B)]
. Le gradient par rapport aux valeurs non vides de B.
Constructeurs et Destructeurs | |
---|---|
SparseAddGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input backprop_val_grad, :: tensorflow::Input a_indices, :: tensorflow::Input b_indices, :: tensorflow::Input sum_indices) |
Attributs publics | |
---|---|
a_val_grad | |
b_val_grad | |
operation |
Attributs publics
a_val_grad
::tensorflow::Output a_val_grad
b_val_grad
::tensorflow::Output b_val_grad
opération
Operation operation
Fonctions publiques
SparseAddGrad
SparseAddGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input backprop_val_grad, ::tensorflow::Input a_indices, ::tensorflow::Input b_indices, ::tensorflow::Input sum_indices )