Build Neural Network Tensorflow Vs Pytorch