MatMul คลาสสุดท้ายสาธารณะ
คูณเมทริกซ์ "a" ด้วยเมทริกซ์ "b"
อินพุตต้องเป็นเมทริกซ์สองมิติ และมิติภายในของ "a" (หลังจากถูกย้ายหาก transpose_a เป็นจริง) จะต้องตรงกับมิติภายนอกของ "b" (หลังจากถูกย้ายแล้ว หาก transposed_b เป็นจริง)
หมายเหตุ : การใช้งานเคอร์เนลเริ่มต้นสำหรับ MatMul บน GPU จะใช้ cublas
คลาสที่ซ้อนกัน
| ระดับ | MatMul.ตัวเลือก | คุณลักษณะทางเลือกสำหรับ MatMul | |
ค่าคงที่
| สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
| เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
| คงที่ <T ขยาย TType > MatMul <T> | สร้าง (ขอบเขต ขอบเขต ตัวดำเนินการ <T> a ตัวดำเนินการ <T> b ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ MatMul ใหม่ |
| เอาท์พุต <T> | ผลิตภัณฑ์ () |
| MatMul.Options แบบคงที่ | transposeA (บูลีนทรานสโพสเอ) |
| MatMul.Options แบบคงที่ | transposeB (ทรานสโพสบูลีน B) |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "MatMul"
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สร้าง MatMul <T> คงที่สาธารณะ (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> a, ตัวดำเนินการ <T> b, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ MatMul ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- ตัวอย่างใหม่ของ MatMul
MatMul.Options แบบคงที่สาธารณะ transposeA (Boolean transposeA)
พารามิเตอร์
| ย้ายA | ถ้าเป็นจริง "a" จะถูกย้ายก่อนการคูณ |
|---|
MatMul.Options สาธารณะคงที่ transposeB (Boolean transposeB)
พารามิเตอร์
| ย้ายB | ถ้าเป็นจริง "b" จะถูกย้ายก่อนการคูณ |
|---|