the shape of elements

[mengjie123]

Hi,

I was trying to run the model, but I failed. It reminded: 'c' argument has 16641 elements, which is inconsistent with 'x' and 'y' with size 16384, on Line 506. I found that the shape of x_star is [128,128], but the shape of u_pred is [129,129]. Could you please check the code?

Best wishes