Neural Network Inference In Pytorch