jax.check_tracer_leaks#

jax.check_tracer_leaks = <jax._src.config.State object>#

jax_check_tracer_leaks config 選項的上下文管理器。

在追蹤完成後立即開啟檢查洩漏的追蹤器。啟用洩漏檢查可能會對效能產生影響:某些快取會被停用,並且可能會增加其他額外負擔。此外,請注意,某些 Python 偵錯器可能會導致誤報,因此建議在啟用洩漏檢查時停用任何偵錯器。

參數:

new_val (Any)