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

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

שיטות בירושה

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

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

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

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

הציבור סטטי CollectivePermute <T> ליצור ( היקף היקף, האופרנד <T> קלט, האופרנד <Integer> sourceTargetPairs)

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

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

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

הקלט המתוקן.