시작하기무료로 시작하기

예측용 RNN 만들기

이제 첫 번째 순환 신경망을 만들어 볼까요? 시퀀스-투-벡터 모델로, 두 개의 레이어를 가진 RNN 레이어와 hidden_size가 32인 구조입니다. RNN 레이어 뒤에는 간단한 선형 레이어가 위치해 출력값을 예측해야 하는 단일 값으로 매핑합니다.

다음 임포트는 이미 완료되어 있습니다:

import torch
import torch.nn as nn

이 연습은 강의의 일부입니다

PyTorch로 배우는 Intermediate Deep Learning

강의 보기

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

class Net(nn.Module):
    def __init__(self):
        super().__init__()
        # Define RNN layer
        self.rnn = ____(
            ____,
            ____,
            ____,
            ____,
        )
        self.fc = nn.Linear(32, 1)
코드 편집 및 실행