jax.numpy.fromiter

内容

jax.numpy.fromiter#

jax.numpy.fromiter(*args, **kwargs)[source]#

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

此函数有意保持未实现,因为它可能是非纯的,因此在 JIT 和其他 JAX 转换中使用是不安全的。考虑改用 jnp.asarray(np.fromiter(...)),但如果 np.fromiter 在 jax 转换中使用,则应注意其可能导致消耗可迭代对象的副作用;有关更多信息,请参见 常见问题:纯函数