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

CrossReplicaSum

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

Op เพื่อรวมอินพุตระหว่างอินสแตนซ์ TPU ที่จำลองแบบ

แต่ละอินสแตนซ์จะป้อนข้อมูลของตนเอง

ตัวอย่างเช่น สมมติว่ามีอินสแตนซ์ TPU 8 รายการ: `[A, B, C, D, E, F, G, H]` ผ่าน group_assignment=`[[0,2,4,6],[1,3,5,7]]` ตั้งค่า `A, C, E, G` เป็นกลุ่ม 0, และ `B, D, F, H` เป็นกลุ่มที่ 1 ดังนั้นเราจึงได้ผลลัพธ์: `[A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+ G, B+D+F+H, A+C+E+G, B+D+F+H]`

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

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

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

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

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

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

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

สาธารณะคง CrossReplicaSum <T> สร้าง ( ขอบเขต ขอบเขต Operand <T> การป้อนข้อมูล ตัวถูกดำเนินการ <Integer> groupAssignment)

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
ป้อนข้อมูล อินพุตท้องถิ่นไปยังผลรวม
groupAssignment เทนเซอร์ int32 ที่มีรูปร่าง [num_groups, num_replicas_per_group] `group_assignment[i]` แทนรหัสจำลองในกลุ่มย่อย ith
คืนสินค้า
  • ตัวอย่างใหม่ของ CrossReplicaSum

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

ผลรวมของอินพุตแบบกระจายทั้งหมด