XlaSplitND

מחלקה גמר ציבורית XlaSplitND

מפצל טנזור קלט על פני כל הממדים.

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

אופציה זו עשויה להיווצר באמצעות גשר ה-TPU.

לדוגמה, עם טנסור 'קלט':

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
`num_splits`:
[2, 2]
ו'ריפודים':
[1, 1]
ה'פלטים' הצפויים הם:
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]

כיתות מקוננות

מַחלָקָה XlaSplitND.Options תכונות אופציונליות עבור XlaSplitND

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

סטטי <T> XlaSplitND <T>
ליצור (היקף היקף , קלט Operand <T>, Long N, List<Long> numSplits, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת XlaSplitND חדשה.
איטרטור< Operand <T>>
איטרטור ()
רשימה< פלט <T>>
פלטים ()
סטטי XlaSplitND.Options
ריפודים (רפידות רשימה<Long>)

שיטות בירושה

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

סטטי ציבורי XlaSplitND <T> ליצור (היקף היקף , קלט Operand <T>, Long N, List<Long> numSplits, Options... אפשרויות)

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

פרמטרים
תְחוּם ההיקף הנוכחי
קֶלֶט קלט טנסור לפיצול על פני כל הממדים. } out_arg { name: "outputs" description: <
numSplits מספר דרכים לפיצול לכל מימד. מידות הצורה חייבות להיות ניתנות לחלוקה שווה.
אפשרויות נושא ערכי תכונות אופציונליות
מחזיר
  • מופע חדש של XlaSplitND

איטרטור ציבורי< Operand <T>> איטרטור ()

רשימה ציבורית< פלט <T>> פלטים ()

רפידות XlaSplitND.Options סטטיות ציבוריות (רפידות רשימה<Long>)

פרמטרים
ריפודים רשימה אופציונלית של ריפודים ימניים לכל מימד של טנזור קלט ליישום לפני הפיצול. זה יכול לשמש כדי להפוך ממד לחלוקה שווה.