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
jumlahSparseTensor
, 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 )