เทนเซอร์โฟลว์ :: ops :: MatrixSetDiag

#include <array_ops.h>

ส่งคืนเมตริกซ์เมตริกซ์แบบแบตช์พร้อมค่าทแยงมุมใหม่

สรุป

เมื่อระบุ 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

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

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

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

operation
output

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

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

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

การดำเนินการ

Operation operation

เอาท์พุท

::tensorflow::Output output

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

MatrixSetDiag

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

โหนด

::tensorflow::Node * node() const 

ตัวดำเนินการ :: tensorflow :: อินพุต

 operator::tensorflow::Input() const 

ตัวดำเนินการ :: tensorflow :: เอาท์พุท

 operator::tensorflow::Output() const