tensoreflusso:: ops:: SparseAddGrad

#include <sparse_ops.h>

L'operatore gradiente per SparseAdd op.

Riepilogo

L'operazione SparseAdd calcola A + B, dove A, B e la somma sono tutti rappresentati come oggetti SparseTensor . Questa operazione accetta il gradiente a monte rispetto ai valori non vuoti della somma e restituisce i gradienti rispetto ai valori non vuoti di A e B.

Argomenti:

  • scope: un oggetto Scope
  • backprop_val_grad: 1-D con forma [nnz(sum)] . Il gradiente rispetto ai valori non vuoti della somma.
  • a_indici: 2-D. Gli indices dello SparseTensor A, dimensione [nnz(A), ndims] .
  • b_indici: 2-D. Gli indices dello SparseTensor B, dimensione [nnz(B), ndims] .
  • somma_indici: 2-D. Gli indices della somma SparseTensor , size [nnz(sum), ndims] .

Ritorna:

  • Output a_val_grad: 1-D con forma [nnz(A)] . Il gradiente rispetto ai valori non vuoti di A.
  • Output b_val_grad: 1-D con forma [nnz(B)] . Il gradiente rispetto ai valori non vuoti di B.

Costruttori e distruttori

SparseAddGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input backprop_val_grad, :: tensorflow::Input a_indices, :: tensorflow::Input b_indices, :: tensorflow::Input sum_indices)

Attributi pubblici

a_val_grad
b_val_grad
operation

Attributi pubblici

a_val_grad

::tensorflow::Output a_val_grad

b_val_grad

::tensorflow::Output b_val_grad

operazione

Operation operation

Funzioni pubbliche

SparseAddGrad

 SparseAddGrad(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input backprop_val_grad,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input b_indices,
  ::tensorflow::Input sum_indices
)