เทนเซอร์โฟลว์ :: ops :: กระจัดกระจาย

#include <sparse_ops.h>

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

สรุป

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

อาร์กิวเมนต์:

  • ขอบเขต: วัตถุ ขอบเขต
  • backprop_val_grad: 1-D ที่มีรูปร่าง [nnz(sum)] การไล่ระดับสีเทียบกับค่าที่ไม่ว่างเปล่าของผลรวม
  • a_indices: 2-D indices ของ SparseTensor A, ขนาด [nnz(A), ndims]
  • b_indices: 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

หน้าที่สาธารณะ

กระจัดกระจาย

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