TensorDiag

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

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

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

`output[i1,..., ik, i1,..., ik] = เส้นทแยงมุม[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]]