jax.Array.committed#

abstract property Array.committed: bool[原始碼]#

陣列是否已提交。

當陣列透過 JAX API 明確放置在裝置上時,即為已提交。例如,jax.device_put(np.arange(8), jax.devices()[0]) 已提交到裝置 0。而 jax.device_put(np.arange(8)) 則為未提交,將放置在預設裝置上。

涉及某些已提交輸入的計算將在已提交的裝置上發生,並且結果也將提交到相同的裝置。在提交到不同裝置的引數上調用運算將引發錯誤。

例如

` a = jax.device_put(np.arange(8), jax.devices()[0]) b = jax.device_put(np.arange(8), jax.devices()[1]) a + b  # 引發錯誤 `

請參閱 https://jax.dev.org.tw/en/latest/faq.html#controlling-data-and-computation-placement-on-devices 以取得更多資訊。