Il tuo primo neural network
È il momento di implementare una piccola rete neurale che contiene due layer lineari in sequenza.
Questo esercizio fa parte del corso
Introduzione al Deep Learning con PyTorch
Istruzioni dell'esercizio
- Aggiungi un contenitore per impilare i layer in sequenza.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
import torch
import torch.nn as nn
input_tensor = torch.Tensor([[2, 3, 6, 7, 9, 3, 2, 1]])
# Create a container for stacking linear layers
model = nn.____(nn.Linear(8, 4),
nn.Linear(4, 1)
)
output = model(input_tensor)
print(output)