jax.Array.device

抽象 属性 Array.device: Device | Sharding

与Array API兼容的设备属性。

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