מוצא ערכים ואינדקסים של הרכיבים '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> | |
פלט < TInt32 > | מדדים () המדדים של 'ערכים' בתוך הממד האחרון של 'קלט'. |
סטטי TopK.Options | ממוין (ממוין בוליאני) |
פלט <T> | ערכים () ה-'k' האלמנטים הגדולים ביותר לאורך כל פרוסה ממדית אחרונה. |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
שיטות ציבוריות
Public static TopK <T> create ( היקף היקף , קלט Operand <T>, Operand < TInt32 > k, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TopK חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
קֶלֶט | 1-D ומעלה עם ממד אחרון לפחות 'k'. |
ק | 0-D. מספר האלמנטים העליונים שיש לחפש לאורך הממד האחרון (לאורך כל שורה עבור מטריצות). |
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של TopK
ציבורי סטטי TopK. אפשרויות ממוינות (ממוין בוליאני)
פרמטרים
מְמוּיָן | אם זה נכון, האלמנטים 'k' שיתקבלו ימוינו לפי הערכים בסדר יורד. |
---|