BroadcastHelper คลาสสุดท้ายสาธารณะ
ตัวช่วยดำเนินการออกอากาศแบบ XLA
ออกอากาศ `lhs` และ `rhs` ไปยังอันดับเดียวกัน โดยเพิ่มขนาด 1 มิติให้กับ `lhs` และ `rhs` ใดก็ตามที่มีอันดับต่ำกว่า โดยใช้กฎการออกอากาศของ XLA สำหรับตัวดำเนินการไบนารี
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
คงที่ <T ขยาย TType > BroadcastHelper <T> | สร้าง (ขอบเขต ขอบเขต ตัวดำเนินการ <T> lhs ตัวดำเนินการ <T> rhs ตัวดำเนินการ <? ขยาย TNumber > BroadcastDims) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ BroadcastHelper ใหม่ |
เอาท์พุต <T> | lhsเอาท์พุต () เทนเซอร์ LHS ที่ออกอากาศ |
เอาท์พุต <T> | rhsเอาท์พุต () เทนเซอร์ RHS ที่ออกอากาศ |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "XlaBroadcastHelper"
วิธีการสาธารณะ
BroadcastHelper แบบคงที่สาธารณะ <T> สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> lhs, ตัวดำเนินการ <T> rhs, ตัวดำเนินการ <? ขยาย TNumber > BroadcastDims)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ BroadcastHelper ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
นะ | เทนเซอร์อินพุต LHS |
Rhs | เทนเซอร์อินพุต RHS |
ออกอากาศ Dims | ข้อกำหนดมิติการออกอากาศแบบ XLA |
การส่งคืน
- อินสแตนซ์ใหม่ของ BroadcastHelper