jax.debug_nans

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

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

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

参数:

new_val (任意类型)