שיעור הגמר הציבורי CrossReplicaSum
אופ לסיכום תשומות על פני מופעי TPU משוכפלים.
כל מופע מספק קלט משלו.
לדוגמה, נניח שיש 8 מופעי TPU: `[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]`.
קבועים
| חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
| פלט <T> | asOutput () מחזירה את הידית הסמלית של הטנזור. |
| סטטי <T מרחיב TNummer > CrossReplicaSum <T> | |
| פלט <T> | פלט () סכום כל התשומות המבוזרות. |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
ערך קבוע: "CrossReplicaSum"
שיטות ציבוריות
פלט ציבורי <T> asOutput ()
מחזירה את הידית הסמלית של הטנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
Public static CrossReplicaSum <T> create ( scope scope, Operand <T> קלט, Operand < TInt32 > groupAssignment)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CrossReplicaSum חדשה.
פרמטרים
| תְחוּם | ההיקף הנוכחי |
|---|---|
| קֶלֶט | הקלט המקומי לסכום. |
| מטלה קבוצתית | טנזור int32 עם צורה [מספר_קבוצות, מספר_עותקים_לקבוצה]. `group_assignment[i]` מייצג את מזהי העתק בתת-הקבוצה ה-ith. |
החזרות
- מופע חדש של CrossReplicaSum