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 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

Ver curso

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