jax.numpy.fromiter

jax.numpy.fromiter(*args, **kwargs)

jnp.fromiter 的未实现 JAX 包装器。

此函数故意保持未实现状态，因为它可能不是纯函数，因此在使用 JIT 和其他 JAX 转换时可能不安全。请考虑改用 jnp.asarray(np.fromiter(...))，但如果 np.fromiter 在 jax 转换中使用，则应谨慎，因为它可能会产生消耗可迭代对象的副作用；有关更多信息，请参阅 JAX 中的常见陷阱：纯函数。