jax.debug_infs

jax.debug_infs = <jax._src.config.State object>

用于 jax_debug_infs 配置选项的上下文管理器。

为每个操作添加无穷大检查。当在 jit 编译计算的输出上检测到无穷大时，调用未编译版本以尝试更精确地识别产生无穷大的操作。

参数：

new_val (Any)