jax.print_environment_info#

jax.print_environment_info(return_string=False)[source]#

返回一个包含本地环境和 JAX 安装信息的字符串。

这是提问或提交错误报告时应包含的有用信息。

参数: return_string (bool) : 如果为 True,则返回字符串而不是打印到标准输出。

参数:

return_string (bool)

返回类型:

str | None