שיעור הגמר הציבורי התאספו
עוטף את מפעיל 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