jax.debug_infs

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

用於 jax_debug_infs 設定選項的情境管理器。

為每個操作新增 inf 檢查。當在即時編譯計算的輸出中偵測到 inf 時，呼叫未編譯版本，以嘗試更精確地識別產生 inf 的操作。

參數:

new_val (Any)