첫 번째 신경망 만들기
이제 두 개의 linear 층을 순차적으로 연결한 작은 신경망을 직접 구현해 봅시다.
이 연습은 강의의 일부입니다
PyTorch로 배우는 딥러닝 입문
연습 안내
- 층을 순서대로 쌓을 수 있는 컨테이너를 추가하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
import torch
import torch.nn as nn
input_tensor = torch.Tensor([[2, 3, 6, 7, 9, 3, 2, 1]])
# Create a container for stacking linear layers
model = nn.____(nn.Linear(8, 4),
nn.Linear(4, 1)
)
output = model(input_tensor)
print(output)