DepthwiseConv2dNative

bookmark_border קל לארגן דפים בעזרת אוספים אפשר לשמור ולסווג תוכן על סמך ההעדפות שלך.

מחשב קונבולוציית עומק דו-ממדית בהינתן טנסור 4-D `קלט` ו`פילטר`.

בהינתן טנסור קלט של צורה `[אצווה, in_height, in_width, in_channels]` ו-filter / kernel tensor של צורה `[filter_height, filter_width, in_channels, channel_multiplier]`, המכיל מסנני פיתול `in_channels` של עומק 1, `depthwise_conv2d` מחיל מסנן אחר על כל ערוץ קלט (מתרחב מערוץ 1 לערוצים 'מכפיל_ערוץ' עבור כל אחד מהם), ולאחר מכן משרשר את התוצאות יחד. לפיכך, לפלט יש ערוצי `in_channels * channel_multiplier`.

for k in 0..in_channels-1
   for q in 0..channel_multiplier-1
     output[b, i, j, k * channel_multiplier + q] =
       sum_{di, dj input[b, strides[1] * i + di, strides[2] * j + dj, k] *
                         filter[di, dj, k, q]
 }