jax.Array.committed

property Array.committed: bool

数组是否已提交。

当通过 JAX API 将数组显式放置到设备上时，该数组被认为是已提交的（committed）。例如，jax.device_put(np.arange(8), jax.devices()[0]) 已提交到设备 0。而 jax.device_put(np.arange(8)) 是未提交的（uncommitted），将被放置在默认设备上。

涉及一些已提交输入的计算将发生在已提交的设备上，并且结果也将提交到相同的设备上。在已提交到不同设备的参数上调用操作将引发错误。

例如

` a = jax.device_put(np.arange(8), jax.devices()[0]) b = jax.device_put(np.arange(8), jax.devices()[1]) a + b  # 引发错误 `

更多信息请参见 https://jax.net.cn/en/latest/faq.html#controlling-data-and-computation-placement-on-devices。