1. 학습
    2. /
  2. 강의
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

연습 문제

예측용 RNN 만들기

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

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

import torch
import torch.nn as nn

지침 1/4

undefined XP
    1
    2
    3
    4
  • input_size, hidden_size, num_layers, batch_first에 올바른 값을 넣어 RNN 레이어를 정의하고 self.rnn에 할당하세요.