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 indices du SparseTensor A, taille [nnz(A), ndims] .
  • b_indices : 2D. Les indices du SparseTensor B, taille [nnz(B), ndims] .
  • sum_indices : 2D. Les indices du SparseTensor de somme, taille [nnz(sum), ndims] .

Retours :

  • Output a_val_grad : 1-D avec la forme [nnz(A)] . Le gradient par rapport aux valeurs non vides de A.
  • Output b_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
)