jax.check_tracer_leaks

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

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

开启检查跟踪器泄漏，一旦跟踪完成就立即进行。启用泄漏检查可能会对性能产生影响：某些缓存将被禁用，并且可能会增加其他开销。此外，请注意，某些 Python 调试器可能会导致误报，因此建议在启用泄漏检查时禁用任何调试器。

参数:

new_val (Any)