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

Retour:

  • 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
)