https://coursevania.com/courses/python-pytorch-programming-with-coding-exercises