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

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

#include <nn_ops.h>

מבצע איגום מקסימלי על הקלט ומפיק הן ערכים מקסימליים והן מדדים.

סיכום

המדדים argmax שטוחים, כך שערך מקסימלי במיקום [b, y, x, c] הופך לאינדקס שטוח: (y * width + x) * channels + c אם include_batch_in_index הוא False; ((b * height + y) * width + x) * channels + c אם include_batch_in_index נכון.

המדדים המוחזרים הם תמיד ב- [0, height) x [0, width) לפני הרידוד, גם אם מדובר בריפוד והתשובה הנכונה המתמטית היא בחוץ (או שלילית או גדולה מדי). זהו באג, אך תיקון קשה לעשות זאת בצורה תואמת לאחור, במיוחד עקב שיטוח.

טענות:

  • היקף: אובייקט Scope
  • קלט: 4-D עם צורה [batch, height, width, channels] . קלט לבריכה.
  • גודל: גודל החלון לכל מימד של טנסור הקלט.
  • צעדים: צעד החלון הזזה לכל מימד של טנסור הקלט.
  • ריפוד: סוג אלגוריתם הריפוד לשימוש.

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

  • include_batch_in_index: האם לכלול ממד אצווה באינדקס שטוח של argmax .

החזרות:

  • Output פלט: טנסור הפלט המרבי המאוחד.
  • Output ארגמאקס: 4-D. המדדים השטוחים של הערכים המקסימליים שנבחרו עבור כל פלט.

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

MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding)
MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs)

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

argmax
operation
output

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

IncludeBatchInIndex (bool x)
Targmax (DataType x)

סטרוקטורים

tensorflow :: ops :: MaxPoolWithArgmax :: Attrs

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

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

ארגמקס

::tensorflow::Output argmax

מבצע

Operation operation

תְפוּקָה

::tensorflow::Output output

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

MaxPoolWithArgmax

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

MaxPoolWithArgmax

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding,
  const MaxPoolWithArgmax::Attrs & attrs
)

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

IncludeBatchInIndex

Attrs IncludeBatchInIndex(
  bool x
)

Targmax

Attrs Targmax(
  DataType x
)