jax.debug_infs

jax.debug_infs = <jax._src.config.State 对象>

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

为每个操作添加 inf 检查。当在 jit 编译的计算输出中检测到 inf 时，会调用未编译的版本，以尝试更精确地确定产生 inf 的操作。

参数:

new_val (任意类型)