MNIST1 20. 간단한 CNN 모델 만들기 import torchimport torch.nn as nnimport torch.optim as optim# 배치크기 * 채널 * 너비 * 높이inputs = torch.Tensor(1,1,28,28)print(inputs.shape)첫번째 Conv2D# Conv2d(입력 데이터가 1개, 출력되는 피쳐의 수 32개, 마스크가 3*3짜리(기울기), padding='same': 테두리를 채워서 크기를 유지)conv1 = nn.Conv2d(in_channels=1, out_channels=32, kernel_size=3, padding='same')out = conv1(inputs)print(out.shape)# 우선 CNN 레이어를 설정하는 과정에서 컨볼루션 연산을 수행합니다.# 여기서 kernel_siz.. 2024. 6. 17. 이전 1 다음
