构建一个用于预测的 RNN
现在是构建您的第一个循环网络的时候了!这是一个序列到向量的模型,包含一个有两层、hidden_size 为 32 的 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)