tensorflow::ops::SparseAddGrad
#include <sparse_ops.h>ตัวดำเนินการไล่ระดับสำหรับการดำเนินการ SparseAdd
สรุป
โอเปอเรเตอร์ SparseAdd คำนวณ A + B โดยที่ A, B และผลรวมทั้งหมดถูกแทนด้วยออบเจ็กต์ SparseTensor โอเปอเรเตอร์นี้รับค่าเกรเดียนต์ต้นทางเทียบกับค่าที่ไม่ว่างเปล่าของผลรวม และส่งออกค่าเกรเดียนต์เทียบกับค่าที่ไม่ว่างเปล่าของ A และ B
ข้อโต้แย้ง:
- ขอบเขต: วัตถุ ขอบเขต
- backprop_val_grad: 1 มิติ รูปทรง
[nnz(sum)]เกรเดียนต์เทียบกับค่าที่ไม่ว่างเปล่าของผลรวม - a_indices: 2 มิติ
indicesของSparseTensorA ขนาด[nnz(A), ndims] - b_indices: 2 มิติ
indicesของSparseTensorB ขนาด[nnz(B), ndims] - sum_indices: 2 มิติ
indicesของผลรวมSparseTensorขนาด[nnz(sum), ndims]
การส่งคืนสินค้า:
-
Outputa_val_grad: มิติเดียวที่มีรูปร่าง[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
หน้าที่สาธารณะ
SparseAddGrad
SparseAddGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input backprop_val_grad, ::tensorflow::Input a_indices, ::tensorflow::Input b_indices, ::tensorflow::Input sum_indices )