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]`.

קבועים

חוּט

OP_NAME

השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

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

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

שיטות בירושה

מכיתה org.tensorflow.op.RawOp

בוליאנית סופית	שווה (Object obj)
int סופי	hashcode ()
מבצע	אופ () החזר יחידת חישוב זו Operation אחת.
מחרוזת סופית	toString ()

מכיתה java.lang.Object

בוליאני	שווה (Object arg0)
שיעור אחרון<?>	getClass ()
int	hashcode ()
ריק סופי	להודיע ​​()
ריק סופי	הודע הכל ()
חוּט	toString ()
ריק סופי	המתן (ארג0 ארוך, int arg1)
ריק סופי	חכה (ארג0 ארוך)
ריק סופי	חכה ()

מתוך ממשק org.tensorflow.op.Op

אבסטרקט ExecutionEnvironment	env () החזר את סביבת הביצוע שבה נוצר האופציה הזו.
מבצע מופשט	אופ () החזר יחידת חישוב זו Operation אחת.

מממשק org.tensorflow.Operand

פלט מופשט <T>	asOutput () מחזירה את הידית הסמלית של הטנזור.
תקציר T	asTensor () מחזיר את הטנזור באופרנד זה.
צורה מופשטת	צורה () מחזירה את הצורה (הידועה אולי בחלקה) של הטנזור שאליו מתייחס Output של אופרנד זה.
מחלקה מופשטת<T>	סוג () מחזירה את סוג הטנזור של אופרנד זה

מהממשק org.tensorflow.ndarray.Shaped

מופשט int	דרגה ()
צורה מופשטת	צורה ()
מופשט ארוך	גודל () מחשב ומחזיר את הגודל הכולל של מיכל זה, במספר ערכים.

קבועים

מחרוזת סופית סטטית ציבורית 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

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

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