Tu primera red neuronal
Es hora de que pongas en marcha una pequeña red neuronal que contenga dos capas lineales en secuencia.
Este ejercicio forma parte del curso
Introducción al aprendizaje profundo con PyTorch
Instrucciones del ejercicio
- Añade un contenedor para apilar capas en secuencia.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
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)