tensorflow:: אופס:: TopK

#include <nn_ops.h>

מוצא ערכים ואינדקסים של 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 .

בנאים והורסים

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)

תכונות ציבוריות

indices
operation
values

פונקציות סטטיות ציבוריות

Sorted (bool x)

מבנים

tensorflow:: ops:: TopK:: Attrs

קובעי תכונות אופציונליים עבור TopK .

תכונות ציבוריות

מדדים

::tensorflow::Output indices

מבצע

Operation operation

ערכים

::tensorflow::Output values

תפקידים ציבוריים

TopK

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

TopK

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

פונקציות סטטיות ציבוריות

מְמוּיָן

Attrs Sorted(
  bool x
)