tensorflow::ops::SparseAddGrad
#include <sparse_ops.h>L'opérateur de gradient 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 par des objets SparseTensor . Cette opération prend en entrée le gradient amont par rapport aux valeurs non vides de la somme et renvoie les gradients par rapport aux valeurs non vides de A et B.
Arguments :
- portée : un objet Scope
- backprop_val_grad : 1-D de forme
[nnz(sum)]. Le gradient par rapport aux valeurs non vides de la somme. - a_indices : 2-D. Les
indicesduSparseTensorA, taille[nnz(A), ndims]. - b_indices : 2D. Les
indicesduSparseTensorB, taille[nnz(B), ndims]. - sum_indices : 2D. Les
indicesduSparseTensorde somme, taille[nnz(sum), ndims].
Retours :
-
Outputa_val_grad : 1-D avec la forme[nnz(A)]. Le gradient par rapport aux valeurs non vides de A. -
Outputb_val_grad : 1-D de 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 )