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

#include <array_ops.h>

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

สรุป

เมื่อกำหนด 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 : เทนเซอร์เอาท์พุท

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

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

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

operation
output

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

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

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

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

Operation operation

เอาท์พุท

::tensorflow::Output output

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

Diag

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

โหนด

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const