您的第一个神经网络
现在请实现一个小型神经网络，其中按顺序包含两个线性层。
本练习是课程的一部分
使用 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)