aliran tensor:: operasi:: SparseAddGrad

#include <sparse_ops.h>

Operator gradien untuk operasi SparseAdd .

Ringkasan

Operasi SparseAdd menghitung A + B, dengan A, B, dan jumlah semuanya direpresentasikan sebagai objek SparseTensor . Operasi ini mengambil gradien hulu dengan nilai penjumlahan yang tidak kosong, dan mengeluarkan gradien dengan nilai A dan B yang tidak kosong.

Argumen:

  • ruang lingkup: Objek Lingkup
  • backprop_val_grad: 1-D dengan bentuk [nnz(sum)] . Gradien terhadap nilai penjumlahan yang tidak kosong.
  • a_indeks: 2-D. indices SparseTensor A, ukuran [nnz(A), ndims] .
  • b_indeks: 2-D. indices SparseTensor B, ukuran [nnz(B), ndims] .
  • jumlah_indeks: 2-D. indices jumlah SparseTensor , ukuran [nnz(sum), ndims] .

Pengembalian:

  • Output a_val_grad: 1-D dengan bentuk [nnz(A)] . Gradien terhadap nilai A yang tidak kosong.
  • Output b_val_grad: 1-D dengan bentuk [nnz(B)] . Gradien terhadap 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

a_val_grad

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