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.

Full content is available with a subscription.
Get full access to all courses on the platform for one year with a single payment.
Unlike other platforms that charge per course, here you get everything for one price, and after one year of use there will be no automatic charge for the following year.