Netwerk met lineaire laag
Neural networks bevatten vaak veel lagen, maar de meeste daarvan zijn lineaire lagen. Als je een enkele lineaire laag begrijpt, snap je hoe ze werken voordat je extra complexiteit toevoegt.
Pas een lineaire laag toe op een input-tensor en bekijk de uitvoer.
Deze oefening maakt deel uit van de cursus
Introductie tot Deep Learning met PyTorch
Oefeninstructies
- Maak een
Linear-laag die 3 features als invoer neemt en 2 outputs teruggeeft. - Laat
input_tensordoor de lineaire laag gaan.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
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)