פיצול `SparseTensor` לטנזורים `num_split` לאורך מימד אחד.
אם ה-`shape[split_dim]` אינו כפולה שלמה של `num_split`. פרוסות `[0 : shape[split_dim] % num_split]` מקבלים ממד אחד נוסף. לדוגמה, אם `split_dim = 1` ו-`num_split = 2` והקלט הוא
input_tensor = shape = [2, 7] [ ade ] [bc ]
מבחינה גרפית טנסורי הפלט הם:
output_tensor[0] = shape = [2, 4] [ a ] [bc ]
output_tensor[1] = shape = [2, 3] [ de ] [ ]
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
סטטי <T מרחיב את TType > SparseSplit <T> | |
רשימה< פלט < TInt64 >> | מדדי פלט () |
רשימה< פלט < TInt64 >> | outputShape () רשימה של טנסור 1-D מייצגת את הצורה של טנסור הפלט הדל. |
רשימה< פלט <T>> | ערכי פלט () רשימה של טנסורים 1-D מייצגת את ערכי הטנזורים הדלילים של הפלט. |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
שיטות ציבוריות
ציבורי סטטי SparseSplit <T> create ( scope scope, Operand < TInt64 > splitDim, Operand < TInt64 > אינדקסים, ערכי Operand <T>, Operand < TInt64 > shape, Long numSplit)
שיטת מפעל ליצירת מחלקה העוטפת פעולת SparseSplit חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
splitDim | 0-D. הממד שלאורכו לפצל. חייב להיות בטווח `[0, rank(shape))`. |
מדדים | טנזור דו-ממדי מייצג את המדדים של הטנזור הדליל. |
ערכים | טנזור 1-D מייצג את ערכי הטנזור הדליל. |
צוּרָה | 1-D. טנסור מייצג את צורת הטנסור הדליל. מדדי פלט: רשימה של טנסור 1-D מייצגת את המדדים של טנסור פלט דליל. |
numSplit | מספר הדרכים לפיצול. |
החזרות
- מופע חדש של SparseSplit