ray.data.from_numpy

ray.data.from_numpy(ndarrays: Union[numpy.ndarray, List[numpy.ndarray]]) → ray.data.dataset.Dataset[ray.data._internal.arrow_block.ArrowRow]

Create a dataset from a list of NumPy ndarrays.

Parameters

ndarrays – A NumPy ndarray or a list of NumPy ndarrays.

Returns

Dataset holding the given ndarrays.

PublicAPI: This API is stable across Ray releases.