Pytorch Building Neural Network From Scratch