שיעור הגמר הציבורי 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 |
שיטות ציבוריות
שיטות בירושה
| בוליאני | שווה (Object arg0) |
| שיעור אחרון<?> | getClass () |
| int | hashcode () |
| ריק סופי | להודיע () |
| ריק סופי | הודע הכל () |
| חוּט | toString () |
| ריק סופי | המתן (ארג0 ארוך, int arg1) |
| ריק סופי | חכה (ארג0 ארוך) |
| ריק סופי | חכה () |
| אבסטרקט ExecutionEnvironment | env () החזר את סביבת הביצוע שבה נוצר האופציה הזו. |
| מבצע מופשט |
קבועים
מחרוזת סופית סטטית ציבורית 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