Linear Regression using PyTorch