CommencerCommencez gratuitement

Votre premier réseau neuronal

Il est temps pour vous de mettre en œuvre un petit réseau neuronal contenant deux couches linéaires successives.

Cet exercice fait partie du cours

<cours>Introduction au Deep Learning avec PyTorch</cours>
Voir le cours

Instructions de l’exercice

  • Ajoutez un conteneur pour empiler les 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)
Modifier et exécuter le code