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

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>
ליצור ( היקף היקף, האופרנד <T> קלט, האופרנד <Integer> groupAssignment)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CrossReplicaSum חדשה.
פלט <T>
פלט ()
סכום כל התשומות המבוזרות.

שיטות בירושה

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

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

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

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

הציבור סטטי CrossReplicaSum <T> ליצור ( היקף היקף, האופרנד <T> קלט, האופרנד <Integer> groupAssignment)

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

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

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

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