Votre premier réseau de neurones
À vous maintenant d'implanter un petit réseau de neurones qui contient deux couches linéaires en séquence.
Cette activité fait partie du cours
Introduction au Deep Learning avec PyTorch
Instructions de l’exercice
- Ajoutez un conteneur pour empiler des couches en séquence.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
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)