jax.export.default_export_platform

jax.export.default_export_platform()

检索默认导出平台。

其中之一：tpu、cpu、cuda、rocm。

返回类型:

str