Description
Tasks
You are required to complete the following tasks:
1. Install pytorch
1. Modify the `rosenbrock.py` file so that it calculates the minimum of the rosenbrock function
1. Upload your completed code to Sakai
Optional
You are not required to complete the following tasks,
however they are good exercises to get you familiar with pytorch.
1. Modify the `rosenbrock` function so that instead of taking two scalar variables as input, it takes a 2-dimensional vector as input.
You will also have to update the optimization code to handle the new type of input.
1. Extend the `rosenbrock` function to arbitrary dimensions using wikipedia’s high-dimensional rosenbrock function: https://en.wikipedia.org/wiki/Rosenbrock_function
## Submission
Upload your `rosenbrock.py` file to sakai
