TopK

כיתת גמר ציבורית TopK

מוצא ערכים ואינדקסים של הרכיבים 'k' הגדולים ביותר עבור הממד האחרון.

אם הקלט הוא וקטור (דרגה-1), מוצא את הערכים 'k' הגדולים ביותר בוקטור ומוציא את הערכים והמדדים שלהם בתור וקטורים. לפיכך, `ערכים[j]` הוא הערך `j` בגודלו ב`קלט`, והאינדקס שלו הוא `מדדים[j]`.

עבור מטריצות (בהתאמה קלט בדרגה גבוהה יותר), מחשב את ערכי 'k' העליונים בכל שורה (בהתאמה וקטור לאורך הממד האחרון). לכן,

values.shape = indices.shape = input.shape[:-1] + [k]

אם שני אלמנטים שווים, האלמנט בעל האינדקס התחתון מופיע ראשון.

כיתות מקוננות

מעמד TopK.Options תכונות אופציונליות עבור TopK

קבועים

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

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

סטטי <T מרחיב TNummer > TopK <T>
ליצור ( היקף היקף , קלט Operand <T>, Operand < TInt32 > k, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TopK חדשה.
פלט < TInt32 >
מדדים ()
המדדים של 'ערכים' בתוך הממד האחרון של 'קלט'.
סטטי TopK.Options
ממוין (ממוין בוליאני)
פלט <T>
ערכים ()
ה-'k' האלמנטים הגדולים ביותר לאורך כל פרוסה ממדית אחרונה.

שיטות בירושה

קבועים

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

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

ערך קבוע: "TopKV2"

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

Public static TopK <T> create ( היקף היקף , קלט Operand <T>, Operand < TInt32 > k, אפשרויות... אפשרויות)

שיטת מפעל ליצירת מחלקה העוטפת פעולת TopK חדשה.

פרמטרים
תְחוּם ההיקף הנוכחי
קֶלֶט 1-D ומעלה עם ממד אחרון לפחות 'k'.
ק 0-D. מספר האלמנטים העליונים שיש לחפש לאורך הממד האחרון (לאורך כל שורה עבור מטריצות).
אפשרויות נושא ערכי תכונות אופציונליות
החזרות
  • מופע חדש של TopK

פלט ציבורי < TInt32 > מדדים ()

המדדים של 'ערכים' בתוך הממד האחרון של 'קלט'.

ציבורי סטטי TopK. אפשרויות ממוינות (ממוין בוליאני)

פרמטרים
מְמוּיָן אם זה נכון, האלמנטים 'k' שיתקבלו ימוינו לפי הערכים בסדר יורד.

ערכי פלט <T> ציבוריים ()

ה-'k' האלמנטים הגדולים ביותר לאורך כל פרוסה ממדית אחרונה.