CrossReplicaSum

อินพุต Op to sum ในอินสแตนซ์ 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]`