Neural Network Dense Pytorch