DepthwiseConv2dNative

bookmark_border Mantenha tudo organizado com as coleções Salve e categorize o conteúdo com base nas suas preferências.

Calcula uma convolução de profundidade 2-D dados os tensores de `entrada` e` filtro` 4-D.

Dado um tensor de entrada de forma `[batch, in_height, in_width, in_channels]` e um tensor de filtro / kernel de forma `[filter_height, filter_width, in_channels, channel_multiplier]`, contendo filtros convolucionais `in_channels` de profundidade 1,` depthwise_conv2d` aplica um filtro diferente a cada canal de entrada (expandindo de 1 canal para canais `channel_multiplier` para cada um) e, a seguir, concatena os resultados juntos. Assim, a saída tem canais `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]
 }