728x90

torch.nn.Conv2d(in_channels, 
		out_channels,
                kernel_size,
                stride=1,
                padding=0,
                dilation=1,
                groups=1,
                bias=True,
                padding_mode='zeros')

is_channel : 이미지의 채널 수 (ex, color dim)

out_channel : Filter의 개수 = Output의 depth

kernel_size : Filter의 크기

padding : Convolution이 진행하면서 겹쳐지는 공간을 0으로 채워 연산 수를 동일하게 해준다.

stride : 필터 진행 간격

반응형
다했다