Diseño del modelo: la clase nn.Module
Construir una red neuronal en PyTorch es como ensamblar LEGO. En _init_ declaras las piezas (capas) y, en adelante, especificas cómo debe fluir el tensor de entrada a través de ellas para convertirse en una predicción. La principal ventaja de nn.Module es que rastrea automáticamente todos los parámetros (pesos) dentro de sus capas.