Added a new solution for assignment 2 which uses the torch nn module.

Shows how the same neural network can be created much easier using the torch nn module.