אופציה להחלפת טנזורים במופעי TPU משוכפלים.
כל מופע מספק קלט משלו.
לדוגמה, נניח שיש 4 מופעי TPU: `[A, B, C, D]`. העברת source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` מקבלת את הפלטים: `[D,A,B,C]`.
שיטות ציבוריות
פלט <T> | asOutput () מחזירה את הידית הסמלית של טנזור. |
סטטי <T> CollectivePermute <T> | |
פלט <T> | פלט () הקלט המתוקן. |
שיטות בירושה
שיטות ציבוריות
פלט ציבורי <T> asOutput ()
מחזירה את הידית הסמלית של טנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
public static CollectivePermute <T> create ( scope scope, Operand <T> input, Operand <Integer> sourceTargetPairs)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CollectivePermute חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
קֶלֶט | הקלט המקומי שיש לשנות. כרגע תומך רק ב-float וב-bfloat16. |
sourceTargetPairs | טנזור עם צורה [מספר_זוגות, 2]. |
החזרות
- מופע חדש של CollectivePermute