אסוף פרוסות מציר `פרמס` לפי `מדדים`.
'מדדים' חייב להיות טנסור שלם של כל מימד (בדרך כלל 0-D או 1-D). מייצר טנזור פלט עם הצורה `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` כאשר:
# Scalar indices (output is rank(params) - 1).
output[a_0, ..., a_n, b_0, ..., b_n] =
params[a_0, ..., a_n, indices, b_0, ..., b_n]
# Vector indices (output is rank(params)).
output[a_0, ..., a_n, i, b_0, ..., b_n] =
params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
# Higher rank indices (output is rank(params) + rank(indices) - 1).
output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
שימו לב שב-CPU, אם נמצא אינדקס מחוץ לתחום, מוחזרת שגיאה. ב-GPU, אם נמצא אינדקס מחוץ לתחום, 0 מאוחסן בערך הפלט המתאים.
ראה גם `tf.batch_gather` ו- tf.gather_nd
.
כיתות מקוננות
מַחלָקָה | אסוף.אפשרויות | תכונות אופציונליות עבור Gather |
שיטות ציבוריות
פלט <T> | asOutput () מחזירה את הידית הסמלית של טנזור. |
סטטי Gather.Options | batchDims (ארוך batchDims) |
סטטי <T, U מרחיב את המספר, V מרחיב את המספר> אסוף <T> | ליצור ( היקף היקף , פרמטרים של Operand <T>, אינדקסים של Operand <U>, ציר Operand <V>, אפשרויות... אפשרויות) שיטת מפעל ליצירת מחלקה העוטפת פעולת Gather חדשה. |
פלט <T> | פלט () ערכים מ-'params' שנאספו ממדדים שניתנו על-ידי 'מדדים', עם הצורה 'params.shape[:axis] + indices.shape + params.shape[ציר + 1:]'. |
שיטות בירושה
שיטות ציבוריות
פלט ציבורי <T> asOutput ()
מחזירה את הידית הסמלית של טנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
סטטי ציבורי אסוף <T> ליצור ( היקף היקף , פרמטרים של Operand <T>, אינדקסים של Operand <U>, ציר Operand <V>, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת Gather חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
params | הטנסור שממנו לאסוף ערכים. חייב להיות לפחות דירוג 'ציר + 1'. |
מדדים | טנזור אינדקס. חייב להיות בטווח `[0, params.shape[axis])`. |
צִיר | הציר ב'פארמים' שממנו לאסוף 'מדדים'. ברירת המחדל למימד הראשון. תומך באינדקסים שליליים. |
אפשרויות | נושא ערכי תכונות אופציונליות |
מחזיר
- מופע חדש של Gather
פלט ציבורי <T> פלט ()
ערכים מ-'params' שנאספו ממדדים שניתנו על-ידי 'מדדים', עם הצורה 'params.shape[:axis] + indices.shape + params.shape[ציר + 1:]'.