jax.print_environment_info# jax.print_environment_info(return_string=False)[源代码]# 返回一个包含本地环境和 JAX 安装信息的字符串。 在提问或提交错误时,包含此信息非常有用。 Args: return_string (bool) : 如果为 True,则返回字符串而不是打印到 stdout。 参数: return_string (bool) 返回类型: str | None