jax.Array.copy_to_host_async

abstract Array.copy_to_host_async()

非同步地將 Array 複製到主機。

對於存在於加速器 (例如 GPU 或 TPU) 上的陣列，JAX 可能會在主機上快取陣列的值。通常，當使用者請求裝置上陣列的值時，這會在幕後發生，但是等到請求值時才啟動裝置到主機的複製，需要 JAX 在等待複製完成時封鎖呼叫者。

copy_to_host_async 請求 JAX 填入陣列的主機上快取，但不等待複製完成。這可能會加速未來對陣列內容的主機上存取。