開始使用免費開始

建立一個預測用 RNN

現在來打造你的第一個循環式網路吧!這會是一個 sequence-to-vector 的模型,包含一個有兩層、hidden_size32 的 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)
編輯並執行程式碼