1. Learn
    2. /
  2. Courses
    3. /
  3. Introduction to Deep Learning with PyTorch
Connected

Exercise

Linear layer network

Neural networks often contain many layers, but most of them are linear layers. Understanding a single linear layer helps you grasp how they work before adding complexity.

Apply a linear layer to an input tensor and observe the output.

Instructions

100 XP
  • Create a Linear layer that takes 3 features as input and returns 2 outputs.
  • Pass input_tensor through the linear layer.