neural_network_for_Pytorch