1. Learn
    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。