jax.export.默认导出平台

jax.export.默认导出平台#

jax.export.default_export_platform()[源代码][源代码]#

检索默认的导出平台。

其中之一:tpucpucudarocm

返回类型:

str