jax.numpy.fromiter

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

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

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