tensorflow::ops::SparseAddGrad
#include <sparse_ops.h>
Оператор градиента для операции SparseAdd.
Краткое описание
Операция SparseAdd вычисляет A + B, где A, B и сумма представлены в виде SparseTensor объектов. Данная операция принимает градиент по непустым значениям суммы и выводит градиенты по непустым значениям A и B.
Аргументы:
- scope: Объект Scope
- backprop_val_grad: 1-мерный массив со структурой
[nnz(sum)]. Градиент по непустым значениям суммы. - a_indices: 2-мерный массив. Индексы непустых значений
SparseTensorA, размер[nnz(A), ndims]. - b_indices: 2-мерный массив. Индексы непустых значений
SparseTensorB, размер[nnz(B), ndims]. - sum_indices: 2-мерный массив. Индексы непустых значений суммы
SparseTensor, размер[nnz(sum), ndims].
Возвращаемые значения:
-
Outputa_val_grad: 1-мерный массив со структурой[nnz(A)]. Градиент по непустым значениям A. -
Outputb_val_grad: 1-мерный массив со структурой[nnz(B)]. Градиент по непустым значениям B.
| Конструкторы и деструкторы | |
|---|---|
SparseAddGrad(const ::tensorflow::Scope & scope, ::tensorflow::Input backprop_val_grad, ::tensorflow::Input a_indices, ::tensorflow::Input b_indices, ::tensorflow::Input sum_indices) |
| Публичные атрибуты | |
|---|---|
a_val_grad | |
b_val_grad | |
operation | |
Публичные атрибуты
a_val_grad
::tensorflow::Output a_val_grad
b_val_grad
::tensorflow::Output b_val_grad
operation
Operation operation
Публичные функции
SparseAddGrad
SparseAddGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input backprop_val_grad, ::tensorflow::Input a_indices, ::tensorflow::Input b_indices, ::tensorflow::Input sum_indices )
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/cc/class/tensorflow/ops/sparse-add-grad