CommencezCommencez gratuitement

Réseau à couche linéaire

Les réseaux de neurones contiennent souvent de nombreuses couches, mais la plupart sont des couches linéaires. Comprendre une seule couche linéaire vous aide à saisir leur fonctionnement avant d'ajouter de la complexité.

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

Cette activité fait partie du cours

Introduction au Deep Learning avec PyTorch

Voir le cours

Instructions de l’exercice

  • Créez une couche Linear qui prend 3 caractéristiques en entrée et retourne 2 sorties.
  • Faites passer 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