Skip to content

voiage.parallel.distributed.chunked_numpy_partition

chunked_numpy_partition([positional or keyword] data: np.ndarray = None, [positional or keyword] partition_count: int = None) -> list[np.ndarray]

Partition a 1D numpy array into balanced views for distributed CPU work.

Parameters:

  • data np.ndarray
  • partition_count int

Returns: list[np.ndarray]