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 )
אלא אם צוין אחרת, התוכן של דף זה הוא ברישיון Creative Commons Attribution 4.0 ודוגמאות הקוד הן ברישיון Apache 2.0. לפרטים, ניתן לעיין במדיניות האתר Google Developers. Java הוא סימן מסחרי רשום של חברת Oracle ו/או של השותפים העצמאיים שלה.
עדכון אחרון: 2024-11-21 (שעון UTC).