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

ส่งกลับเทนเซอร์แนวทแยงด้วยค่าแนวทแยงที่กำหนด

สรุป

เมื่อกำหนด diagonal การดำเนินการนี้จะส่งคืนเทนเซอร์ที่มี diagonal และทุกอย่างเสริมด้วยศูนย์ เส้นทแยงมุมคำนวณดังนี้:

สมมติว่า diagonal มีขนาด [D1,..., Dk] ดังนั้นเอาต์พุตจะเป็นเทนเซอร์อันดับ 2k โดยมีขนาด [D1,..., Dk, D1,..., Dk] โดยที่:

output[i1,..., ik, i1,..., ik] = diagonal[i1, ..., ik] และ 0 ทุกที่

ตัวอย่างเช่น:

# 'diagonal' is [1, 2, 3, 4]
tf.diag(diagonal) ==> [[1, 0, 0, 0]
                       [0, 2, 0, 0]
                       [0, 0, 3, 0]
                       [0, 0, 0, 4]]

ข้อโต้แย้ง:

• ขอบเขต: วัตถุ ขอบเขต
• เส้นทแยงมุม: อันดับ k เทนเซอร์ โดยที่ k มีค่าสูงสุด 1

ผลตอบแทน:

• Output : เทนเซอร์เอาท์พุต

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

Operation operation

::tensorflow::Output output

 Diag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input diagonal
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const