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>	פלט () הקלט המתוקן.

שיטות בירושה

מכיתה 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

ערך קבוע: "CollectivePermute"

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

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

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

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

public static CollectivePermute <T> create ( scope scope, Operand <T> input, Operand < TInt32 > sourceTargetPairs)

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

פרמטרים

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

החזרות

• מופע חדש של CollectivePermute

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

הקלט המתוקן.