你的第一個神經網路
現在該由你來實作一個小型的神經網路，依序包含兩個 linear 層。
本練習屬於課程
使用 PyTorch 的深度學習入門
練習說明
- 新增一個容器，用來依序堆疊各層。
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
import torch
import torch.nn as nn
input_tensor = torch.Tensor([[2, 3, 6, 7, 9, 3, 2, 1]])
# Create a container for stacking linear layers
model = nn.____(nn.Linear(8, 4),
nn.Linear(4, 1)
)
output = model(input_tensor)
print(output)