jax.Array.device

property Array.device: Device | Sharding

Array API 兼容的设备属性。

对于单设备数组，这将返回 Device。对于分片数组，这将返回 Sharding。