when input is simple, the result is also not good

[Calsia]

when i run test.py, the script print when i sent just a 'p(32,40,24,20)' to the model, the model return p(32,40,24,20)u(24,40,32,16)+u(32,16,32,28) so,what should i focus on to fix this ? thanks for any response