Yeah, I think I haven’t mastered how to properly use grids.
I’m trying to train a NN that can learn f(x,y), so for each pair of x,y there is a value associated with it.
This specific function:
T = np.sin(3*X) * np.sin(3*Y)

I wanted to do this example first to try and see if I can do it before using it for more complicated inputs such as images. That is why I tried to use np.mgrid  to produce sample (x,y) uniformly and use it as my training set. Could you think of any way I could do it more easily?