jax.export.default_export_platform

jax.export.default_export_platform()

检索默认导出平台。

以下之一： 'tpu', 'cpu', 'cuda', 'rocm'。

返回类型:

str