TracerArrayConversionError: The numpy.ndarray conversion method __array__() was called on traced array with shape float32[10000].

[ciri1998]

I am trying to modify Ben Moseley's code available on github https://github.com/benmoseley/FBPINNs. My intention is to insert a vector of values into the loss fn that is dependent on x y coordinates, and I need the original vector Z to be interpolated as a function of x and y, and then the values at the same coordinates with which the algorithm samples x and y are extracted, so that the values match. The problem I have encountered is that within loss fn I cannot use libraries other than JAX and to my knowledge there are no functions within JAX to interpolate in 2D.

I'm trying to get around the problem in every way but I'm not succeeding, one of my ideas was to extrapolate the x,y points sampled by the algorithm but I'm not succeeding, the code is really very articulated. Would anyone be able to give me any advice/help on this?

There would be the function jax.scipy.ndimage.map_coordinates but it doesn't work properly and the points it extrapolates are meaningless, but could it work?

[dfm]

I'm not very familiar with this specific use case, but if your x and y points are on a regular grid, perhaps you could use jax.scipy.interpolate.RegularGridInterpolator (source). I noticed that this doesn't seem to be included in the generated API docs, but it should be fine to use it if it works for you!