Python Code For Gradient Descent In Deep