tensorflow::ops::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] for 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