עזרה להגן על שונית המחסום הגדולה עם TensorFlow על Kaggle הצטרפו אתגר

זרימת טנסור :: אופ :: TopK

#include <nn_ops.h>

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

סיכום

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

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

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

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

טענות:

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

תכונות אופציונליות (ראה Attrs ):

  • מיון: אם נכון, אלמנטים 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
)