RaggedGather

שיעור הגמר הציבורי 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)` הוא טנסור הפלט.
(הערה: אופציה זו של c++ משמשת ליישום ה-python 'tf.ragged.gather' ברמה גבוהה יותר, שתומך גם במדדים מרופטים.)

קבועים

חוּט OP_NAME השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

שיטות ציבוריות

סטטי <T מרחיב את TNumber , U מרחיב את TType > RaggedGather <T, U>
create ( scope scope, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <? מרחיב את המדדים TNumber >, Long OUTPUTRAGGEDRANK)
שיטת מפעל ליצירת מחלקה העוטפת פעולת RaggedGather חדשה.
פלט <U>
outputDenseValues ​​()
ה-'flat_values' עבור RaggedTensor המוחזר.
רשימה< פלט <T>>
outputNestedSplits ()
הטנסורים `מפוצלים_שורות` שמגדירים את חלוקת השורות עבור ה-RaggedTensor המוחזר.

שיטות בירושה

קבועים

מחרוזת סופית סטטית ציבורית OP_NAME

השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

ערך קבוע: "RaggedGather"

שיטות ציבוריות

ציבורי סטטי RaggedGather <T, U> create ( היקף היקף , Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <? מרחיב את המדדים Tnumber >, 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

פלט ציבורי <U> outputDenseValues ​​()

ה-'flat_values' עבור RaggedTensor המוחזר.

רשימה ציבורית< פלט <T>> outputNestedSplits ()

הטנסורים 'מפוצלים_שורות' המגדירים את חלוקת השורות עבור ה-RaggedTensor המוחזר.