tensorflow:: ops:: Diag

ส่งคืนเทนเซอร์แนวทแยงที่มีค่าแนวทแยงที่กำหนดให้

สรุป

เมื่อกำหนด 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