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

#include <sparse_ops.h>

ตัวดำเนินการไล่ระดับสีสำหรับ 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

ตัวสร้างและผู้ทำลาย

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

งานสาธารณะ

เบาบางAddGrad 

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