CommencerCommencer 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

Introduction au deep learning avec PyTorch

Afficher le cours

Instructions

  • 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 cet exemple de code.

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