ComenzarEmpieza gratis

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 aumentar la complejidad.

Aplica una capa lineal a un tensor de entrada y observa la salida.

Este ejercicio forma parte del curso

Introducción al aprendizaje profundo con PyTorch

Ver curso

Instrucciones del ejercicio

  • Crea una capa Linear que tome 3 características como entrada y devuelva 2 salidas.
  • Pasa input_tensor por la capa lineal.

Ejercicio interactivo práctico

Prueba este ejercicio completando el código de muestra.

import torch
import torch.nn as nn

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 the linear layer
output = ____(____)

print(output)
Editar y ejecutar código