jax.debug_infs

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

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

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

参数:

new_val (任意类型)