CrossReplicaSum

שיעור הגמר הציבורי 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]`.

שיטות ציבוריות

פלט <T>
asOutput ()
מחזירה את הידית הסמלית של טנזור.
סטטי <T מרחיב מספר> CrossReplicaSum <T>
צור ( scope scope, Operand <T> קלט, Operand <Integer> groupAssignment)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CrossReplicaSum חדשה.
פלט <T>
פלט ()
סכום כל התשומות המבוזרות.

שיטות בירושה

שיטות ציבוריות

פלט ציבורי <T> asOutput ()

מחזירה את הידית הסמלית של טנזור.

כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.

Public static CrossReplicaSum <T> create ( scope scope, Operand <T> input, Operand <Integer> groupAssignment)

שיטת מפעל ליצירת מחלקה העוטפת פעולת CrossReplicaSum חדשה.

פרמטרים
תְחוּם ההיקף הנוכחי
קֶלֶט הקלט המקומי לסכום.
מטלה קבוצתית טנזור int32 עם צורה [מספר_קבוצות, מספר_עותקים_לקבוצה]. `group_assignment[i]` מייצג את מזהי העתק בתת-הקבוצה ה-ith.
החזרות
  • מופע חדש של CrossReplicaSum

פלט ציבורי <T> פלט ()

סכום כל התשומות המבוזרות.