CommencerCommencez gratuitement

Réseau à couches linéaires

Les réseaux neuronaux contiennent souvent de nombreuses couches, mais la plupart d’entre elles sont des couches linéaires. La compréhension d’une couche linéaire unique vous permet de mieux appréhender leur fonctionnement avant d’y ajouter de la complexité.

Appliquez une couche linéaire à un tenseur d’entrée et observez la sortie.

Cet exercice fait partie du cours

<cours>Introduction au deep learning avec PyTorch</cours>
Voir le cours

Instructions de l’exercice

  • Créez une couche Linear qui prend 3 caractéristiques en entrée et renvoie 2 sorties.
  • Passez input_tensor à travers la couche linéaire.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

import torch
import torch.nn as nn

input_tensor = torch.tensor([[0.3471, 0.4547, -0.2356]])

# Create a Linear layer
linear_layer = nn.____(
                         in_features=____, 
                         out_features=____
                         )

# Pass input_tensor through the linear layer
output = ____(____)

print(output)
Modifier et exécuter le code