1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorchによる中級ディープラーニング
Connected

演習

予測用RNNを構築する

いよいよ最初のリカレントネットワークを作成します！ここではシーケンスからベクトルへのモデルを実装します。hidden_size が 32、層数が2の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 に代入します。