1. Learn
    2. /
  2. 课程
    3. /
  3. PyTorch 深度学习进阶
Connected

道练习

构建一个用于预测的 RNN

现在是构建您的第一个循环网络的时候了！这是一个序列到向量的模型，包含一个有两层、hidden_size 为 32 的 RNN 层。RNN 层之后，用一个简单的线性层将输出映射到要预测的单个数值。

以下导入已为您完成：

import torch
import torch.nn as nn

说明 1 / 共 4 个

undefined XP
    1
    2
    3
    4
  • 定义 RNN 层，并为 input_size、hidden_size、num_layers 和 batch_first 传入正确的取值，然后赋给 self.rnn。