שיעור הגמר הציבורי התאספו
עוטף את מפעיל XLA Gather שתועד ב
https://www.tensorflow.org/xla/operation_semantics#gather
קבועים
| חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
| פלט <T> | asOutput () מחזירה את הידית הסמלית של הטנזור. |
| סטטי <T מרחיב את TType , U מרחיב את TNummer > אסוף <T> | |
| פלט <T> | פלט () |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
ערך קבוע: "XlaGather"
שיטות ציבוריות
פלט ציבורי <T> asOutput ()
מחזירה את הידית הסמלית של הטנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
סטטי ציבורי אסוף <T> ליצור ( היקף היקף , אופרנד <T> אופרנד, אינדיסי התחלה של Operand <U>, פרוסות אופרנד <U> פרוסות, מספרי מחרוזת ממדים, מדדים בוליאנייםAreSorted)
שיטת מפעל ליצירת מחלקה העוטפת פעולת Gather חדשה.
פרמטרים
| תְחוּם | ההיקף הנוכחי |
|---|---|
| אופרנד | המערך ממנו אנו מתאספים. |
| מדדי start | מערך המכיל את מדדי ההתחלה של הפרוסות שאנו אוספים. |
| פרוסות גדלים | slice_sizes[i] הוא הגבולות לפרוסה בממד i. |
| dimensionNumbers | פרוטו xla::GatherDimensionNumbers בסידרה. |
| indicesAreSorted | בוליאנית המציינת אם המדדים ממוינים. |
החזרות
- מופע חדש של Gather