jax.numpy.fromiter

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

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

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