NcclReduce

คลาสสุดท้ายสาธารณะ NcclReduce

ลด "อินพุต" จาก "num_devices" โดยใช้ "การลด" ลงในอุปกรณ์เครื่องเดียว

ลด "อินพุต" จาก "num_devices" โดยใช้ "การลด" ลงในอุปกรณ์เครื่องเดียว

ควรสร้างกราฟเพื่อให้อินพุตทั้งหมดมีการกำหนดอุปกรณ์ที่ถูกต้อง และ op เองก็ถูกกำหนดหนึ่งในอุปกรณ์เหล่านี้

อินพุต: อินพุตเพื่อลด ข้อมูล: ค่าที่ลดลงในอุปกรณ์ `num_devices` ทั้งหมด การลดลง: การดำเนินการลดที่จะดำเนินการ

ค่าคงที่

สตริง OP_NAME ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

วิธีการสาธารณะ

เอาท์พุต <T>
เป็นเอาท์พุต ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
คง <T ขยาย TNumber > NcclReduce <T>
create (ขอบเขต ขอบเขต , Iterable< Operand <T>> อินพุต, การลดสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ NcclReduce ใหม่
เอาท์พุต <T>

วิธีการสืบทอด

ค่าคงที่

สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME

ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

ค่าคงที่: "NcclReduce"

วิธีการสาธารณะ

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

ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์

อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต

สร้าง NcclReduce <T> แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , อินพุต < ตัวดำเนินการ <T>>, การลดสตริง)

วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ NcclReduce ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
การส่งคืน
  • อินสแตนซ์ใหม่ของ NcclReduce

ข้อมูล เอาท์พุท สาธารณะ <T> ()