מחלקה סופית ציבורית ReadVariableXlaSplitND
מפצל טנזור קלט של משתנה משאב על פני כל הממדים.
הפעלה שמפצלת את טנזור הקלט של משתנה המשאב בהתבסס על התכונה num_splits הנתונה, משטחת פרוסות באופן אופציונלי ומחזירה את הפרוסות. הפרוסות מוחזרות בסדר עיקרי בשורה.
אופציה זו עשויה להיווצר באמצעות גשר ה-TPU.
לדוגמה, עם טנסור 'קלט':
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[2, 2]
[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
ריפודים ציבוריים סטטיים ReadVariableXlaSplitND.Options (רפידות רשימה<Long>)
פרמטרים
ריפודים | רשימה אופציונלית של ריפודים ימניים לכל מימד של טנזור קלט ליישום לפני הפיצול. זה יכול לשמש כדי להפוך ממד לחלוקה שווה. |
---|