jax.numpy.fromiter#
- jax.numpy.fromiter(*args, **kwargs)[source]#
Unimplemented JAX wrapper for jnp.fromiter.
This function is left deliberately unimplemented because it may be non-pure and thus unsafe for use with JIT and other JAX transformations. Consider using
jnp.asarray(np.fromiter(...))instead, although care should be taken ifnp.fromiteris used within jax transformations because of its potential side-effect of consuming the iterable object; for more information see Common Gotchas: Pure Functions.