เทนเซอร์โฟลว์:: ปฏิบัติการ:: 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