NcclBroadcast

คลาสสุดท้ายสาธารณะ 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

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