Pytorch Machine Learning Example