TensorDiagPart

ส่งกลับส่วนเส้นทแยงมุมของเมตริกซ์

การดำเนินการนี้ส่งคืนเทนเซอร์ที่มีส่วน "แนวทแยง" ของ "อินพุต" ส่วน "แนวทแยง" คำนวณได้ดังนี้:

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

`เส้นทแยงมุม[i1,..., ik] = อินพุต[i1, ..., ik, i1,..., ik]`

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

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