קונצץ טנזור קלט בכל הממדים.
אופ שממזג פורס את טנסור הקלט על סמך התכונה num_splits הנתונה, מפשיט ריפודים באופן אופציונלי ומחזיר את הטנזור הממוזג ללא ריפודים.
אופציה זו עשויה להיווצר באמצעות גשר ה-TPU.
לדוגמה, עם טנסור 'קלט':
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
[2, 2]
[1, 1]
[[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
כיתות מקוננות
מַחלָקָה | XlaConcatND.Options | תכונות אופציונליות עבור XlaConcatND |
שיטות ציבוריות
פלט <T> | asOutput () מחזירה את הידית הסמלית של טנזור. |
סטטי <T> XlaConcatND <T> | צור ( היקף היקף , כניסות לביצוע< Oprand <T>>, רשימה<Long> numConcats, אפשרויות... אפשרויות) שיטת מפעל ליצירת מחלקה העוטפת פעולת XlaConcatND חדשה. |
פלט <T> | פלט () |
סטטי XlaConcatND.Options | ריפודים (רפידות רשימה<Long>) |
שיטות בירושה
שיטות ציבוריות
פלט ציבורי <T> asOutput ()
מחזירה את הידית הסמלית של טנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת לקבלת ידית סמלית המייצגת את חישוב הקלט.
ציבורי סטטי XlaConcatND <T> ליצור (היקף היקף , כניסות איטריות< Oprand <T>>, רשימה<Long> numConcats, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת XlaConcatND חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
תשומות | הזן פרוסות טנזור בסדר שורה ראשי כדי להתמזג בכל הממדים. כל הכניסות חייבות להיות באותה צורה. } out_arg { שם: תיאור "פלט": < |
numConcats | מספר דרכים למיזוג לכל מימד. |
אפשרויות | נושא ערכי תכונות אופציונליות |
מחזיר
- מופע חדש של XlaConcatND
ריפודי XlaConcatND.Options סטטיים ציבוריים (רפידות רשימה<Long>)
פרמטרים
ריפודים | רשימה אופציונלית של ריפודים נכונים לכל מימד להורדה מהטנזור הסופי הממוזג. ריפודים אלה אינם חייבים לחרוג מגודל הממדים של התוצאה הממוזגת לפני הפשטת ריפודים. |
---|