ComenzarEmpieza gratis

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

Ver curso

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)
Editar y ejecutar código