jax.local_device_count

jax.local_device_count(backend=None)

返回此进程可寻址的设备数量。

参数:

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

返回类型:

int