tensorflow::ops::SparseAddGrad

#include <sparse_ops.h>

Operator gradien untuk operasi SparseAdd .

Ringkasan

Operasi SparseAdd menghitung A + B, di mana A, B, dan jumlahnya semuanya direpresentasikan sebagai objek SparseTensor . Operasi ini menerima gradien hulu terhadap nilai-nilai jumlah yang tidak kosong, dan menghasilkan gradien terhadap nilai-nilai A dan B yang tidak kosong.

Argumen:

  • Cakupan: Objek Cakupan
  • backprop_val_grad: 1-D dengan bentuk [nnz(sum)] . Gradien terhadap nilai-nilai non-kosong dari jumlah tersebut.
  • a_indices: 2-D. indices SparseTensor A, ukuran [nnz(A), ndims] .
  • b_indices: 2-D. indices SparseTensor B, ukuran [nnz(B), ndims] .
  • sum_indices: 2-D. indices dari jumlah SparseTensor , ukuran [nnz(sum), ndims] .

Mengembalikan:

  • Output a_val_grad: 1-D dengan bentuk [nnz(A)] . Gradien terhadap nilai-nilai A yang tidak kosong.
  • Output b_val_grad: 1-D dengan bentuk [nnz(B)] . Gradien terhadap nilai-nilai B yang tidak kosong.

Konstruktor dan Destruktor

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

Atribut publik

a_val_grad
b_val_grad
operation

Atribut publik

nilai_nilai

::tensorflow::Output a_val_grad

b_val_grad

::tensorflow::Output b_val_grad

operasi

Operation operation

Fungsi publik

SparseAddGrad

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