@Ingenieur
Can you provide sample inputs where there is a learnable function that maps input to target?
I created an example notebook for running this code:
  
      

      colab.research.google.com
  

  
    

Google Colaboratory 2