ReadVariableXlaSplitND

מחלקה סופית ציבורית ReadVariableXlaSplitND

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

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

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

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

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

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

מעמד ReadVariableXlaSplitND.Options תכונות אופציונליות עבור ReadVariableXlaSplitND

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

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

שיטות בירושה

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

סטטי ציבורי ReadVariableXlaSplitND <T> ליצור ( היקף היקף , משאב Operand <?>, Class<T> T, Long N, List<Long> numSplits, Options... options)

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

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

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

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

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

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