Ditt första neurala nätverk
Nu är det dags att implementera ett litet neuralt nätverk med två linjära lager i sekvens.
Den här övningen är en del av kursen
Introduktion till djupinlärning med PyTorch
Övningsinstruktioner
- Lägg till en behållare för att stapla lager i sekvens.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
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)