שיעור הגמר הציבורי RaggedGather
אסוף פרוסות מרופטות מציר '0' של 'פארמס' לפי 'מדדים'.
מוציא פלט 'RaggedTensor' המורכב מ-'output_dense_values' ו-'output_nested_splits', כך:
output.shape = indices.shape + params.shape[1:]
output.ragged_rank = indices.shape.ndims + params.ragged_rank
output[i...j, d0...dn] = params[indices[i...j], d0...dn]
- `params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)` מספק את הערכים שיש לאסוף.
- `מדדים` הוא טנסור צפוף עם dtype `int32` או `int64`, המציין אילו ערכים יש לאסוף.
- `output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)` הוא טנסור הפלט.
שיטות ציבוריות
סטטי <T מרחיב את מספר, U, V מרחיב את המספר> RaggedGather <T, U> | |
פלט <U> | outputDenseValues () ה-'flat_values' עבור RaggedTensor המוחזר. |
רשימה< פלט <T>> | outputNestedSplits () הטנסורים `מפוצלים_שורות` שמגדירים את חלוקת השורות עבור ה-RaggedTensor המוחזר. |
שיטות בירושה
שיטות ציבוריות
סטטי ציבורי RaggedGather <T, U> create ( היקף היקף , Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, מדדי Operand <V>, Long OUTPUTRAGGEDRANK)
שיטת מפעל ליצירת מחלקה העוטפת פעולת RaggedGather חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
paramsNestedSplits | הטנסורים `nested_row_splits` המגדירים את חלוקת השורות עבור הקלט RaggedTensor `params`. |
paramsDenseValues | ה-'flat_values' עבור ה-'params' RaggedTensor. היה שינוי בטרמינולוגיה ברמת הפיתון מ-dense_values ל-flat_values, כך שה-dense_values הוא השם שהוצא משימוש. |
מדדים | מדדים במימד החיצוני ביותר של `פארמים` של הערכים שיש לאסוף. |
OUTPUTRAGGEDRANK | הדרגה המרופטת של הפלט RaggedTensor. `output_nested_splits` יכיל את המספר הזה של טנזורים `row_splits`. ערך זה צריך להיות שווה ל-'indices.shape.ndims + params.ragged_rank - 1'. |
מחזיר
- מופע חדש של RaggedGather
רשימה ציבורית< פלט <T>> outputNestedSplits ()
הטנסורים `מפוצלים_שורות` שמגדירים את חלוקת השורות עבור ה-RaggedTensor המוחזר.