คลาสสุดท้ายสาธารณะ NcclBroadcast
ส่ง 'อินพุต' ไปยังอุปกรณ์ทั้งหมดที่เชื่อมต่อกับเอาต์พุต
ส่ง 'อินพุต' ไปยังอุปกรณ์ทั้งหมดที่เชื่อมต่อกับเอาต์พุต
ควรสร้างกราฟเพื่อให้ ops ทั้งหมดที่เชื่อมต่อกับเอาต์พุตมีการกำหนดอุปกรณ์ที่ถูกต้อง และ op เองได้รับการกำหนดหนึ่งในอุปกรณ์เหล่านี้
input: อินพุตสำหรับการออกอากาศ เอาท์พุท: เช่นเดียวกับอินพุต รูปร่าง: รูปร่างของเทนเซอร์อินพุต
ค่าคงที่
| สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
| เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
| คง <T ขยาย TNumber > NcclBroadcast <T> | สร้าง (ขอบเขต ขอบเขต , อินพุต Operand <T>, รูปร่างรูปร่าง ) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ NcclBroadcast ใหม่ |
| เอาท์พุต <T> | เอาท์พุท () |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "NcclBroadcast"
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สร้าง NcclBroadcast <T> แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , อินพุต Operand <T>, รูปร่าง รูปร่าง )
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ NcclBroadcast ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|
การส่งคืน
- ตัวอย่างใหม่ของ NcclBroadcast