jax.check_tracer_leaks# jax.check_tracer_leaks = <jax._src.config.State object># jax_check_tracer_leaks 配置选项的上下文管理器。 在跟踪完成后立即打开对泄漏跟踪器的检查。启用泄漏检查可能会对性能产生影响:一些缓存被禁用,并且可能会添加其他开销。此外,请注意,一些 Python 调试器可能会导致误报,因此建议在启用泄漏检查时禁用任何调试器。 参数: new_val (Any)