ส่งกลับเทนเซอร์แนวทแยงด้วยค่าแนวทแยงที่กำหนด
เมื่อพิจารณาจาก "เส้นทแยงมุม" การดำเนินการนี้จะส่งคืนเทนเซอร์ที่มี "เส้นทแยงมุม" และทุกอย่างที่เหลือเสริมด้วยศูนย์ เส้นทแยงมุมคำนวณดังนี้:
สมมติว่า `เส้นทแยงมุม` มีขนาด [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]]
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
คงที่ <T ขยาย TType > TensorDiag <T> | |
เอาท์พุต <T> | เอาท์พุท () |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
วิธีการสาธารณะ
เอาท์พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สร้าง TensorDiag <T> คงที่สาธารณะ ( ขอบเขต ขอบเขต ตัว ดำเนินการ <T> เส้นทแยงมุม)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TensorDiag ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
เส้นทแยงมุม | อันดับ k เทนเซอร์ โดยที่ k มีค่ามากที่สุด 1 |
การส่งคืน
- ตัวอย่างใหม่ของ TensorDiag