jax._src.core.check_scalar_conversion

[lfaucheux]

Is there a specific reason why jax._src.core.check_scalar_conversion relies on ~.ndim instead of, say, ~.size ?

Why prevent, for instance, float conversion of arrays that can still be converted to scalars using ~.item() ?

[jakevdp]

Thanks for the question! JAX strives as much as possible to semantically match the latest release of NumPy. We added this particular code after NumPy deprecated scalar conversions for arrays that are not zero-dimensional; see numpy/numpy#10615.