Hi Santiago,
Check this blog, if it helps.
  

      Medium – 15 Dec 18
  

  
    

Simple Regression with Neural Networks in PyTorch

  Neural networks are sometimes described as a ‘universal function approximator’. Here I show a few examples of simple and slightly more…

  
    Reading time: 2 min read
  

  

  
    
    
  

  


In your other post, what is happening in the “class” function? Can you try with a sequential network with multiple hidden layers with non-linear activations?