Prima ta rețea neuronală
Este momentul să implementezi o rețea neuronală simplă, formată din două straturi liniare dispuse în succesiune.
Acest exercițiu face parte din cursul
Introducere în Deep Learning cu PyTorch
Instrucțiuni pentru exercițiu
- Adaugă un container pentru stivuirea straturilor în succesiune.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
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)