DepthwiseConv2dNative

bookmark_border 컬렉션을 사용해 정리하기 내 환경설정을 기준으로 콘텐츠를 저장하고 분류하세요.

4차원 '입력' 및 '필터' 텐서를 사용하여 2차원 깊이별 컨볼루션을 계산합니다.

'[batch, in_height, in_width, in_channels]' 형태의 입력 텐서와 깊이 1의 'in_channels' 컨벌루션 필터를 포함하는 형태의 '[filter_height, filter_width, in_channels, Channel_multiplier]' 필터/커널 텐서, 'lengthwise_conv2d' 각 입력 채널에 서로 다른 필터를 적용한 다음(각 채널에 대해 1채널에서 `channel_multiplier` 채널로 확장) 결과를 함께 연결합니다. 따라서 출력에는 '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]
 }