jax.Array.device

属性 Array.device: Device | Sharding

兼容 Array API 的设备属性。

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