neural network function recebe size como parametro tb