Reference - https://docs.pytorch.org/tutorials/beginner/blitz/autograd_tutorial.html
Reference