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	:: tensorflow::Output
operation	Operation
values	:: tensorflow::Output

פונקציות סטטיות ציבוריות
Sorted (bool x)	Attrs

מבנים
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
)