jax.local_device_count

jax.local_device_count(backend=None)

傳回此進程可定址的裝置數量。

參數:

backend (str | xla_client.Client | None | None)

回傳類型:

int