jax.local_device_count# jax.local_device_count(backend=None)[源代码][源代码]# 返回此进程可寻址的设备数量。 参数: backend (str | xla_client.Client | None) 返回类型: int
