tensorflow:: ops:: TopK

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

תַקצִיר

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

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

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

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

טיעונים:

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

מאפיינים אופציונליים (ראה Attrs ):

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

החזרות:

• ערכי Output : k האלמנטים הגדולים ביותר לאורך כל פרוסה ממדית אחרונה.
• מדדי Output : מדדי values בתוך הממד האחרון של input .

מאפיינים ציבוריים

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

Attrs Sorted(
  bool x
)