jax.Array.copy_to_host_async

abstract Array.copy_to_host_async()

异步地将 Array 复制到主机。

对于驻留在加速器（如 GPU 或 TPU）上的数组，JAX 可能会在主机上缓存数组的值。通常，当用户请求设备上数组的值时，这种情况会在幕后发生，但是等待到请求值时才启动设备到主机的复制，需要 JAX 在等待复制完成时阻塞调用者。

copy_to_host_async 请求 JAX 填充其数组的主机缓存，但不等待复制完成。这可以加速将来对数组内容的在主机上的访问。