jax.debug_nans

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

用於 jax_debug_nans 設定選項的上下文管理器。

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

參數:

new_val (Any)