Restez organisé à l'aide des collections Enregistrez et classez les contenus selon vos préférences.

tensorflow :: opérations :: SparseAddGrad

#include <sparse_ops.h>

L'opérateur de gradient pour l'opération SparseAdd .

Résumé

L'op SparseAdd calcule A + B, où A, B et la somme sont tous représentés sous SparseTensor objets SparseTensor . Cette opération prend le gradient amont pour les valeurs non vides de la somme, et produit les gradients avec les 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, taille [nnz(A), ndims] .
  • b_indices: 2D. Les indices du SparseTensor B, taille [nnz(B), ndims] .
  • somme_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
)