CollectivePermute

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

אופציה להחלפת טנזורים במופעי 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>
צור ( scope scope, Operand <T> קלט, Operand <Integer> sourceTargetPairs)
שיטת מפעל ליצירת מחלקה העוטפת פעולת CollectivePermute חדשה.
פלט <T>
פלט ()
הקלט המתוקן.

שיטות בירושה

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

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

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

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

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

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

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

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

הקלט המתוקן.