ray.data.from_numpy

ray.data.from_numpy(ndarrays: numpy.ndarray | List[numpy.ndarray]) → MaterializedDataset

从NumPy ndarray列表创建一个 Dataset。

示例

>>> import numpy as np
>>> import ray
>>> arr = np.array([1])
>>> ray.data.from_numpy(arr)
MaterializedDataset(num_blocks=1, num_rows=1, schema={data: int64})

从NumPy数组列表创建一个Ray数据集。

>>> ray.data.from_numpy([arr, arr])
MaterializedDataset(num_blocks=2, num_rows=2, schema={data: int64})

参数:

ndarrays – 一个 NumPy ndarray 或一个 NumPy ndarray 列表。

返回:

Dataset 持有来自给定 ndarrays 的数据。