jax.live_arrays

内容

jax.live_arrays#

jax.live_arrays(platform=None)[source]#

返回 platform 后端中的所有活动数组。

如果 platform 为 None,则为默认后端。