[파이토치] nn.functional.avg_pool2d 관련

 

 

 

 

 

파이토치에서 average pooling에 대한 함수를 제공해주고 있다.

 

average pooling의 경우는 너무 깊은 CNN 차원에 따른 over fitting을 대응하기 위해서 사용되는 기법이다.

 

pooling layer 단에서 사용할 수 있는 기법은 크게 2가지 타입이 있는데, 첫번째로 Max pooling layer, 두번째로 Average pooling layer 가 있다.

• Max pooling layer는 kernel_size에서 표현되는 픽셀 중 가장 최대값을 뽑아낸다.
• Average pooling layer는 kernel_size에서 표현되는 픽셀들의 평균을 뽑아낸다.

 

관련 파라미터를 살펴보도록 하겠다.

• input : 텐서가 들어가는 부분이다. (minibatch, input_channels, height, width)
• kernel_size : 풀링할 window의 사이즈를 정의하는 부분이다.
• stride : window를 얼마만큼 움직일지 결정한다. 보통 kernel_size만큼 이루어지기 때문에, 설정하지 않으면 기본적으로 kernel_size로 책정된다.
• padding : zero padding을 줄지 말지 결정하는 부분이다. 기본은 0으로 잡혀있다.

 

https://pytorch.org/docs/stable/nn.functional.html

torch.nn.functional — PyTorch master documentation