Einsum คลาสสุดท้ายสาธารณะ
op ที่รองรับ einsum op พื้นฐานพร้อม 2 อินพุตและ 1 เอาต์พุต
op นี้มีประสิทธิภาพ TPU ที่ดีกว่า เนื่องจากไม่มีการปรับรูปร่างและดำเนินการย้ายอย่างชัดเจนเหมือนกับที่ tf.einsum ทำ
ค่าคงที่
| สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
| เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
| คงที่ <T ขยาย TType > Einsum <T> | สร้าง ( ขอบเขต ขอบเขต ตัวดำเนินการ <T> a ตัวถูกดำเนินการ <T> b สมการสตริง) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ Einsum ใหม่ |
| เอาท์พุต <T> | ผลิตภัณฑ์ () |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "XlaEinsum"
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สร้าง Einsum สาธารณะ <T> คงที่ (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> a, ตัวดำเนินการ <T> b, สมการสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ Einsum ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|
การส่งคืน
- ตัวอย่างใหม่ของ Einsum