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 )
© 2022 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/api_docs/cc/class/tensorflow/ops/sparse-add-grad