PyTorch: torch.nn.MaxPool2d

danbibibi·2022년 1월 26일
0

PyTorch 🔥

목록 보기
8/20

torch.nn.MaxPool2d

Max Pooling 연산을 위한 Layer
( Pooling: feature map의 크기를 적당히 줄이며, 특정 feature를 강조, 채널 수 변화 x )

사용법

torch.nn.MaxPool2d(kernel_size, stride=None, 
		  		 padding=0, dilation=1, return_indices=False, 
        	  	 ceil_mode=False)

kernel_size: 커널 사이즈 (int or tuple)
stride: stride(몇 칸씩 이동할지) 사이즈 (int or tuple, 기본 값 = 1)
padding: zero padding 사이즈 (int or tuple, 기본 값 = 0)

profile
블로그 이전) https://danbibibi.tistory.com

0개의 댓글