מחלקה סופית ציבורית 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>)
פרמטרים
ריפודים | רשימה אופציונלית של ריפודים נכונים לכל מימד להורדה מהטנזור הסופי הממוזג. ריפודים אלה אינם חייבים לחרוג מגודל הממדים של התוצאה הממוזגת לפני הפשטת ריפודים. |
---|