updated the signature of NeuralNetworkVectorized