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 以取得更多資訊。