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

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

#include <nn_ops.h>

מבצע ריפוד כתהליך מקדים במהלך קונבולוציה.

סיכום

בדומה ל- FusedResizeAndPadConv2d, אפשרות זו מאפשרת יישום אופטימלי שבו שלב טרנספורמציית הריפוד המרחבי מתמזג עם בדיקת im2col, אך במקרה זה ללא הסינון הדו-שנתי הנדרש לצורך שינוי גודל. איחוי הריפוד מונע את הצורך לכתוב את תוצאות הביניים כטנסורים שלמים, ולהפחית את לחץ הזיכרון, ונוכל להשיג רווחי זמן אחזור על ידי מיזוג חישובי הטרנספורמציה. התכונה data_format עבור Conv2D אינה נתמכת על ידי אופ זה, ובמקום זה נעשה שימוש בסדר 'NHWC'. באופן פנימי אופ זה משתמש במאגר שריטות יחיד לכל גרף, מה שאומר שהוא יחסום אם מספר גרסאות מופעלות במקביל. הסיבה לכך היא שמפעיל זה מהווה בעיקר אופטימיזציה למזעור השימוש בזיכרון.

טענות:

  • היקף: אובייקט Scope
  • קלט: 4-D עם צורה [batch, in_height, in_width, in_channels] .
  • ריפודים: מטריצה ​​בת שני עמודים המציינת את גדלי הריפוד. מספר השורות חייב להיות זהה לדרגת input .
  • פילטר: 4-D עם צורה [filter_height, filter_width, in_channels, out_channels] .
  • צעדים: 1-D באורך 4. הצעד של חלון ההזזה לכל מימד input . חייב להיות באותו סדר כמו הממד שצוין בתבנית.
  • ריפוד: סוג אלגוריתם הריפוד לשימוש.

החזרות:

  • Output : טנסור הפלט.

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

FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding)

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

operation
output

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

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

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

מבצע

Operation operation

תְפוּקָה

::tensorflow::Output output

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

FusedPadConv2D

 FusedPadConv2D(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input paddings,
  ::tensorflow::Input filter,
  StringPiece mode,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

צוֹמֶת

::tensorflow::Node * node() const 

אופרטור :: זורם tensor :: קלט

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const