voiage.core.gpu_acceleration.array_to_cpu

array_to_cpu

array_to_cpu([positional or keyword] arr: object = None, [positional or keyword] backend: str | None = None) -> np.ndarray

Transfer an array from GPU memory back to CPU (NumPy).

Args: arr: Array on GPU (type depends on backend) backend: GPU backend that was used (‘jax’, ‘cupy’, ‘torch’, or None for auto-detect)

Returns

NumPy array on CPU

Parameters:

• arr object
• backend str | None (default: None)

Returns: np.ndarray