TensorDiag

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]]
 

ค่าคงที่

สตริง OP_NAME ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

วิธีการสาธารณะ

เอาท์พุต <T>
เป็นเอาท์พุต ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
คงที่ <T ขยาย TType > TensorDiag <T>
สร้าง (ขอบเขต ขอบเขต ตัวดำเนินการ <T> เส้นทแยงมุม)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TensorDiag ใหม่
เอาท์พุต <T>

วิธีการสืบทอด

ค่าคงที่

สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME

ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

ค่าคงที่: "Diag"

วิธีการสาธารณะ

เอาท์ พุท สาธารณะ <T> asOutput ()

ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์

อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต

สร้าง TensorDiag <T> คงที่สาธารณะ ( ขอบเขต ขอบเขต ตัวดำเนินการ <T> เส้นทแยงมุม)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TensorDiag ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
เส้นทแยงมุม อันดับ k เทนเซอร์ โดยที่ k มีค่ามากที่สุด 1
การส่งคืน
  • ตัวอย่างใหม่ของ TensorDiag

เอาท์พุท สาธารณะ <T> เอาท์พุท ()