שיעור הגמר הציבורי 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> | |
| פלט <T> | פלט () הקלט המתוקן. |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית 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