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

演習

GRU ネットワーク

LSTM と並んで広く使われている再帰型ニューラルネットワークの一種が Gated Recurrent Unit（GRU）です。魅力はそのシンプルさにあり、GRU セルは LSTM セルより計算量が少ない一方で、しばしば同等の性能を発揮します。

ここでは、これまでに作成した RNN モデル定義のコードが与えられています。これを GRU ネットワークになるように書き換えてください。torch と torch.nn as nn はすでにインポートされています。

指示

100 XP
  • RNN のモデル定義を更新して GRU ネットワークを作成し、GRU レイヤーを self.gru に代入してください。