เทนเซอร์โฟลว์:: ปฏิบัติการ:: วินิจฉัย
#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( const ::tensorflow::Scope & scope, ::tensorflow::Input diagonal )
โหนด
::tensorflow::Node * node() const
ตัวดำเนินการ::tensorflow::อินพุต
operator::tensorflow::Input() const
ตัวดำเนินการ::tensorflow::เอาต์พุต
operator::tensorflow::Output() const