建立一個預測用 RNN
現在來打造你的第一個循環式網路吧!這會是一個 sequence-to-vector 的模型,包含一個有兩層、hidden_size 為 32 的 RNN。接在 RNN 之後的是一個簡單的線性層,會把輸出對應到要預測的單一數值。
以下匯入已經幫你完成:
import torch
import torch.nn as nn
本練習屬於課程
Intermediate Deep Learning with PyTorch
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
class Net(nn.Module):
def __init__(self):
super().__init__()
# Define RNN layer
self.rnn = ____(
____,
____,
____,
____,
)
self.fc = nn.Linear(32, 1)