dòng chảy căng:: ôi:: thưa thớtThêmGrad

#include <sparse_ops.h>

Toán tử gradient cho tùy chọn SparseAdd .

Bản tóm tắt

Op SparseAdd tính toán A + B, trong đó A, B và tổng đều được biểu diễn dưới dạng đối tượng SparseTensor . Op này lấy các giá trị gradient ngược dòng ghi không trống của tổng và xuất ra các gradient ghi các giá trị không trống của A và B.

Lập luận:

  • phạm vi: Một đối tượng Phạm vi
  • backprop_val_grad: 1-D có hình dạng [nnz(sum)] . Độ dốc đối với các giá trị không trống của tổng.
  • a_chỉ số: 2-D. indices của SparseTensor A, size [nnz(A), ndims] .
  • b_chỉ số: 2-D. indices của SparseTensor B, size [nnz(B), ndims] .
  • tổng_chỉ số: 2-D. indices của tổng SparseTensor , size [nnz(sum), ndims] .

Trả về:

  • Output a_val_grad: 1-D có hình dạng [nnz(A)] . Độ dốc đối với các giá trị khác trống của A.
  • Output b_val_grad: 1-D có hình dạng [nnz(B)] . Độ dốc đối với các giá trị khác trống của B.

Hàm tạo và hàm hủy

SparseAddGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input backprop_val_grad, :: tensorflow::Input a_indices, :: tensorflow::Input b_indices, :: tensorflow::Input sum_indices)

Thuộc tính công khai

a_val_grad
b_val_grad
operation

Thuộc tính công khai

a_val_grad

::tensorflow::Output a_val_grad

b_val_grad

::tensorflow::Output b_val_grad

hoạt động

Operation operation

Chức năng công cộng

thưa thớtThêmGrad

 SparseAddGrad(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input backprop_val_grad,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input b_indices,
  ::tensorflow::Input sum_indices
)