jax.Array.copy_to_host_async

Array.copy_to_host_async()

将 Array 异步复制到主机。

对于位于加速器（如 GPU 或 TPU）上的数组，JAX 可能会在宿主（host）上缓存该数组的值。通常，当用户请求设备上数组的值时，此操作会在后台自动进行；但如果等到请求值时才启动设备到宿主的拷贝，JAX 将必须阻塞调用者以等待拷贝完成。

copy_to_host_async 会请求 JAX 填充其宿主端的数组缓存，但不会等待拷贝完成。这可以加速后续对数组内容的宿主端访问。