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

ส่งกลับเมทริกซ์เทนเซอร์แบบแบทช์พร้อมค่าแนวทแยงแบบแบทช์ใหม่

สรุป

เมื่อระบุ input และ diagonal การดำเนินการนี้จะส่งคืนเทนเซอร์ที่มีรูปร่างและค่าเดียวกันกับ input ยกเว้นเส้นทแยงมุมหลักของเมทริกซ์ด้านในสุด สิ่งเหล่านี้จะถูกเขียนทับด้วยค่าใน diagonal

ผลลัพธ์จะถูกคำนวณดังนี้:

สมมติว่า input มีมิติ k+1 [I, J, K, ..., M, N] และ diagonal มีมิติ k [I, J, K, ..., min(M, N)] จากนั้นเอาต์พุตจะเป็นเทนเซอร์อันดับ k+1 ที่มีขนาด [I, J, K, ..., M, N] โดยที่:

• output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] สำหรับ m == n
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] สำหรับ m != n

ข้อโต้แย้ง:

• ขอบเขต: วัตถุ ขอบเขต
• อินพุต: อันดับ k+1 โดยที่ k >= 1
• เส้นทแยงมุม: อันดับ k โดยที่ k >= 1

ผลตอบแทน:

• Output : อันดับ k+1 โดยมี output.shape = input.shape

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

Operation operation

::tensorflow::Output output

 MatrixSetDiag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input diagonal
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const