יום הקהילה ML הוא 9 בנובמבר! הצטרפו אלינו עדכונים מ- TensorFlow, JAX, ועוד למידע נוסף

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

#include <nn_ops.h>

מציאת ערכים של נתון הסדר ה- n לממד האחרון.

סיכום

אם הקלט הוא וקטור (דרגה 1), מצא את הערכים שהוא הערך ה- n- הקטן ביותר בווקטור ומוציא את ערכיהם כמתח סקלרי.

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

values.shape = input.shape[:-1]

טענות:

  • היקף: אובייקט Scope
  • קלט: 1-D ומעלה עם הממד האחרון לפחות n+1 .
  • n: 0-D. מיקום הווקטור הממוין לבחירה לאורך הממד האחרון (לאורך כל שורה למטריצות). הטווח [0, input.shape[:-1]) של n הוא [0, input.shape[:-1])

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

  • הפוך: כאשר מוגדר כ- True, מצא את הערך התשיעי בגודל הווקטור ולהיפך.

החזרות:

  • Output : נתון הסדר ה- n לאורך כל פרוסת ממד אחרונה.

קונסטרוקטורים ומשחתנים

NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n)
NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n, const NthElement::Attrs & attrs)

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

operation
values

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

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

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

Reverse (bool x)

סטרוקטורים

tensorflow :: ops :: NthElement :: Attrs

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

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

מבצע

Operation operation

ערכים

::tensorflow::Output values

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

NthElement

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n
)

NthElement

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n,
  const NthElement::Attrs & attrs
)

צוֹמֶת

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const 

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

לַהֲפוֹך

Attrs Reverse(
  bool x
)