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.
indicesSparseTensorA, ukuran[nnz(A), ndims]. - b_indices: 2-D.
indicesSparseTensorB, ukuran[nnz(B), ndims]. - sum_indices: 2-D.
indicesdari jumlahSparseTensor, ukuran[nnz(sum), ndims].
Mengembalikan:
-
Outputa_val_grad: 1-D dengan bentuk[nnz(A)]. Gradien terhadap nilai-nilai A yang tidak kosong. -
Outputb_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 )