update a variable in torch code example

Example: update a variable in torch

w1.data -= learning_rate * w1.grad.data
w2.data -= learning_rate * w2.grad.data