DepthwiseConv2dNative

bookmark_border Restez organisé à l'aide des collections Enregistrez et classez les contenus selon vos préférences.

Calcule une convolution en profondeur 2D à partir des tenseurs « entrée » et « filtre » 4-D.

Étant donné un tenseur d'entrée de forme `[batch, in_height, in_width, in_channels]` et un tenseur de filtre/noyau de forme `[filter_height, filter_width, in_channels, channel_multiplier]`, contenant des filtres convolutifs `in_channels` de profondeur 1, `deepwise_conv2d` applique un filtre différent à chaque canal d'entrée (en passant de 1 canal aux canaux `channel_multiplier` pour chacun), puis concatène les résultats ensemble. Ainsi, la sortie a des canaux `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]
 }