Can model predict multiplying?

I’m trying to make a model that can predict that y = sqr(x)^2 or multiplying two numbers of input like y = x1 * x2 with a lot of data of multiplying two numbers but the model can’t learn or predict.
I read in some pages that I need to use something like log input and exp output of numbers before the model and after prediction, I think it may work but I want to make it inside the model so that I can predict more complex functions with multi-input with different operations like summation, multiplication powers and roots in the same function for multi-input
how to build that model :\

I suggest to take a look at:

https://arxiv.org/abs/2101.09530

2 Likes