przepływ tensorowy:: ops:: SparseAddGrad
#include <sparse_ops.h>
Operator gradientu dla operacji SparseAdd .
Streszczenie
Operacja SparseAdd oblicza A + B, gdzie A, B i suma są reprezentowane jako obiekty SparseTensor
. Ta operacja pobiera gradient poprzedzający z niepustymi wartościami sumy i wyprowadza gradienty z niepustymi wartościami A i B.
Argumenty:
- zakres: Obiekt Scope
- backprop_val_grad: 1-D z kształtem
[nnz(sum)]
. Gradient względem niepustych wartości sumy. - a_indeksy: 2-D.
indices
SparseTensor
A, size[nnz(A), ndims]
. - b_indeksy: 2-D.
indices
SparseTensor
B, size[nnz(B), ndims]
. - suma_indices: 2-D.
indices
sumySparseTensor
, size[nnz(sum), ndims]
.
Zwroty:
-
Output
a_val_grad: 1-D z kształtem[nnz(A)]
. Gradient względem niepustych wartości A. -
Output
b_val_grad: 1-D z kształtem[nnz(B)]
. Gradient względem niepustych wartości B.
Konstruktory i destruktory | |
---|---|
SparseAddGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input backprop_val_grad, :: tensorflow::Input a_indices, :: tensorflow::Input b_indices, :: tensorflow::Input sum_indices) |
Atrybuty publiczne | |
---|---|
a_val_grad | |
b_val_grad | |
operation |
Atrybuty publiczne
a_val_grad
::tensorflow::Output a_val_grad
b_val_grad
::tensorflow::Output b_val_grad
działanie
Operation operation
Funkcje publiczne
SparseAddGrad
SparseAddGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input backprop_val_grad, ::tensorflow::Input a_indices, ::tensorflow::Input b_indices, ::tensorflow::Input sum_indices )