开始使用免费开始使用

构建一个用于预测的 RNN

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

以下导入已为您完成:

import torch
import torch.nn as nn

本练习是课程的一部分

PyTorch 深度学习进阶

查看课程

交互式实操练习

通过完成这段示例代码来试试这个练习。

class Net(nn.Module):
    def __init__(self):
        super().__init__()
        # Define RNN layer
        self.rnn = ____(
            ____,
            ____,
            ____,
            ____,
        )
        self.fc = nn.Linear(32, 1)
编辑并运行代码