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
Instructions de l’exercice
- Créez une couche
Linearqui 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)