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

AllToAll

AllToAll המעמד הסופי הציבור

אופ להחלפת נתונים על פני העתקים של TPU.

בכל עותק, הקלט מפוצל לבלוקים של `split_count` לאורך `split_dimension` ונשלח לעותקים האחרים שקיבלו group_assignment. לאחר קבלת `split_count` - 1 בלוקים מעתקים אחרים, אנו משרשרים את הבלוקים לאורך `concat_dimension` כפלט.

לדוגמה, נניח שיש 2 העתקים של TPU: עותק 0 מקבל קלט: `[[A, B]]` עותק 1 מקבל קלט: `[[C, D]]`

group_assignment=`[[0, 1]]` concat_dimension=0 split_dimension=1 split_count=2

הפלט של העתק 0: `[[A], [C]]` הפלט של העתק 1: `[[B], [D]]`

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

פלט <T>
asOutput ()
מחזירה את הידית הסמלית של טנזור.
סטטי <T> AllToAll <T>
ליצור ( היקף היקף, האופרנד <T> קלט, האופרנד <Integer> groupAssignment, לונג concatDimension, לונג splitDimension, לונג splitCount)
שיטת מפעל ליצירת מחלקה העוטפת פעולת AllToAll חדשה.
פלט <T>
פלט ()
התוצאה שהוחלפה.

שיטות בירושה

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

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

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

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

הציבור סטטי AllToAll <T> ליצור ( היקף היקף, האופרנד <T> קלט, האופרנד <Integer> groupAssignment, לונג concatDimension, לונג splitDimension, לונג splitCount)

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

פרמטרים
תְחוּם ההיקף הנוכחי
קֶלֶט הקלט המקומי לסכום.
משימה קבוצתית טנזור int32 עם צורה [מספר_קבוצות, מספר_עותקים_לקבוצה]. `group_assignment[i]` מייצג את מזהי העתק בתת-הקבוצה ה-ith.
concatDimension מספר הממד שיש לשרשר.
splitDimension מספר הממד שיש לפצל.
splitCount מספר הפיצולים, מספר זה חייב להיות שווה לגודל תת-הקבוצה(group_assignment.get_shape()[1])
החזרות
  • מופע חדש של AllToAll

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

התוצאה שהוחלפה.