tensorflow:: אופס:: TopK

bookmark_border קל לארגן דפים בעזרת אוספים אפשר לשמור ולסווג תוכן על סמך ההעדפות שלך.

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

תַקצִיר

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

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

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

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

טיעונים:

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

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

• sorted: אם נכון, ה- 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
)