คำเตือน: API นี้เลิกใช้งานแล้วและจะถูกลบออกใน TensorFlow เวอร์ชันในอนาคตหลังจาก การแทนที่ เสถียรแล้ว

NcclReduce

ชั้นสุดท้ายประชาชน NcclReduce

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

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

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

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

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

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

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

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

ประชาชน เอาท์พุท <T> asOutput ()

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

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

สาธารณะคง NcclReduce <T> สร้าง ( ขอบเขต ขอบเขต Iterable < Operand <T >> อินพุตลด String)

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

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

ประชาชน เอาท์พุท <T> ข้อมูล ()