เทนเซอร์โฟลว์:: ปฏิบัติการ:: เบาบางAddGrad

ตัวดำเนินการไล่ระดับสีสำหรับ SparseAdd op

สรุป

op SparseAdd คำนวณ A + B โดยที่ A, B และผลรวมทั้งหมดแสดงเป็นวัตถุ SparseTensor การดำเนินการนี้ใช้การไล่ระดับสีอัพสตรีมด้วยค่าที่ไม่ว่างเปล่าของผลรวม และส่งออกการไล่ระดับสีด้วยค่าที่ไม่ว่างเปล่าของ A และ B

ข้อโต้แย้ง:

• ขอบเขต: วัตถุ ขอบเขต
• backprop_val_grad: 1-D ที่มีรูปร่าง [nnz(sum)] การไล่ระดับสีเทียบกับค่าที่ไม่ว่างของผลรวม
• a_ดัชนี: 2-D indices ของ SparseTensor A, ขนาด [nnz(A), ndims]
• b_ดัชนี: 2-D indices ของ SparseTensor B ขนาด [nnz(B), ndims]
• sum_indices: 2-D indices ของผลรวม SparseTensor ขนาด [nnz(sum), ndims]

ผลตอบแทน:

• Output a_val_grad: 1-D ที่มีรูปร่าง [nnz(A)] การไล่ระดับสีเทียบกับค่าที่ไม่ว่างของ A
• Output b_val_grad: 1-D ที่มีรูปร่าง [nnz(B)] เกรเดียนต์เทียบกับค่าที่ไม่ว่างของ B

คุณลักษณะสาธารณะ

::tensorflow::Output a_val_grad

::tensorflow::Output b_val_grad

Operation operation

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