jax.interpreters.xla._DeviceArray back to numpy-array

[Anselmoo]

Dear jax-developer and community,

thx for accelerating array operations.

I am wondering if there is an option to switch back the type from DeviceArray back to Numpy, like

jnp.asarray().to_numpy()

or so.

That would be helpful to avoid raise crashes in KERAS:

TypeError: Expect the data to ImageInput to be numpy.ndarray or tf.data.Dataset, 
but got <class 'jaxlib.xla_extension.DeviceArray'>.

[mattjj]

Thanks for the question!

Just call np.array(jax_array) where np is from import numpy as np.

Does that answer your question?

[Anselmoo]

Tested and checked @mattjj

thx, it works!