AssignVariableXlaConcatND

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

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

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

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

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

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
 
`num_splits`:
[2, 2]
 
ו-`paddings`:
[1, 1]
 
ה`פלטים` הצפויים הם:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

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

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

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

סטטי <T> AssignVariableXlaConcatND
ליצור ( היקף היקף, משאב Operand <?>, כניסות איטריות < Operand <T>>, רשימה<Long> numConcats, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת AssignVariableXlaConcatND חדשה.
static AssignVariableXlaConcatND.Options
ריפודים (רפידות רשימה<Long>)

שיטות בירושה

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

public static AssignVariableXlaConcatND create ( היקף היקף , משאב Operand <?>, כניסות איטריות < Operand <T>>, רשימה<Long> numConcats, אפשרויות... אפשרויות)

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

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

public static AssignVariableXlaConcatND.Options ריפודים (רפידות רשימה<Long>)

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