DepthwiseConv2dNative

یک انحراف عمقی دو بعدی را با توجه به تانسورهای 4 بعدی «ورودی» و «فیلتر» محاسبه می‌کند.

با توجه به تانسور ورودی شکل «[batch, in_height, in_width, in_channels]» و یک فیلتر / تانسور هسته شکل «[filter_height, filter_width, in_channels, channel_multiplier]» که حاوی فیلترهای کانولوشنال «in_channels» با عمق 1_ «depthwidse» است. فیلتر متفاوتی را برای هر کانال ورودی اعمال می‌کند (از 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]
 }