כיתת גמר ציבורית ResourceGather
אסוף פרוסות מהמשתנה שעליו הצביע `משאב` לפי `מדדים`.
'מדדים' חייב להיות טנסור שלם של כל מימד (בדרך כלל 0-D או 1-D). מייצר טנזור פלט עם הצורה `indices.shape + params.shape[1:]` כאשר:
# Scalar indices
output[:, ..., :] = params[indices, :, ... :]
# Vector indices
output[i, :, ..., :] = params[indices[i], :, ... :]
# Higher rank indices
output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]
כיתות מקוננות
מַחלָקָה | ResourceGather.Options | תכונות אופציונליות עבור ResourceGather |
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
פלט <U> | asOutput () מחזירה את הידית הסמלית של הטנזור. |
סטטי ResourceGather.Options | batchDims (ארוך batchDims) |
סטטי <U מרחיב את TType > ResourceGather <U> | create ( scope scope, Operand <?> resource, Operand <? מרחיב את TNomber > indexen, Class<U> dtype, Options... options) שיטת מפעל ליצירת מחלקה העוטפת פעולת ResourceGather חדשה. |
פלט <U> | פלט () |
סטטי ResourceGather.Options | validateIndices (בוליאנית validateIndices) |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
ערך קבוע: "ResourceGather"
שיטות ציבוריות
פלט ציבורי <U> asOutput ()
מחזירה את הידית הסמלית של הטנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
public static ResourceGather <U> create ( scope scope, Operand <?> resource, Operand <? מרחיב את המדדים של TNumber , Class<U> dtype, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת ResourceGather חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
אפשרויות | נושא ערכי תכונות אופציונליות |
מחזיר
- מופע חדש של ResourceGather