Red de capas lineales
Las redes neuronales suelen contener muchas capas, pero la mayoría son capas lineales. Comprender una sola capa lineal te ayuda a entender cómo funcionan antes de añadirles complejidad. En este ejercicio, aplicarás una capa lineal a un tensor de entrada y observarás la salida.
torch.nn
(nn
), y torch
ya están importados para ti.
Este ejercicio forma parte del curso
Introducción al aprendizaje profundo con PyTorch
Instrucciones de ejercicio
- Crea una capa lineal que tome 3 características como entrada y devuelva 2 salidas.
- Pasa
input_tensor
por la capa lineal.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
input_tensor = torch.tensor(
[[0.3471, 0.4547, -0.2356]])
# Create a linear layer
linear_layer = nn.____(
in_features=____,
out_features=____
)
# Pass input_tensor through linear layer
output = ____
print(output)