CollectivePermute

שיעור הגמר הציבורי CollectivePermute

אופציה להחלפת טנזורים במופעי TPU משוכפלים.

כל מופע מספק קלט משלו.

לדוגמה, נניח שיש 4 מופעי TPU: `[A, B, C, D]`. העברת source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` מקבלת את הפלטים: `[D,A,B,C]`.

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

פלט <T>	asOutput () מחזירה את הידית הסמלית של הטנזור.
סטטי <T מרחיב את TType > CollectivePermute <T>	צור ( scope scope, Operand <T> קלט, Operand < TInt32 > sourceTargetPairs) שיטת מפעל ליצירת מחלקה העוטפת פעולת CollectivePermute חדשה.
פלט <T>	פלט () הקלט המתוקן.

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

מכיתה java.lang.Object

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

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

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

קבועים

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

ערך קבוע: "CollectivePermute"

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

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

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

תְחוּם	ההיקף הנוכחי
קֶלֶט	הקלט המקומי שיש לשנות. כרגע תומך רק ב-float וב-bfloat16.
sourceTargetPairs	טנזור עם צורה [מספר_זוגות, 2].

הקלט המתוקן.