jax.Array.device

内容

jax.Array.device#

abstract property Array.device: Device | Sharding[source]#

Array API 兼容的设备属性。

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